perbaikan beberapa fitur
This commit is contained in:
parent
9db6afa110
commit
bbdf79124f
|
|
@ -13,12 +13,12 @@ import json
|
||||||
class RiceLeafClassificationAPI:
|
class RiceLeafClassificationAPI:
|
||||||
"""Client untuk Rice Leaf Disease Classification API"""
|
"""Client untuk Rice Leaf Disease Classification API"""
|
||||||
|
|
||||||
def __init__(self, base_url: str = "http://127.0.0.1:8000/api/classification"):
|
def __init__(self, base_url: str = "http://127.0.0.1:5000"):
|
||||||
"""
|
"""
|
||||||
Initialize API client
|
Initialize API client
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
base_url: Base URL untuk API endpoint
|
base_url: Base URL untuk API endpoint (default: Flask API on port 5000)
|
||||||
"""
|
"""
|
||||||
self.base_url = base_url
|
self.base_url = base_url
|
||||||
self.session = requests.Session()
|
self.session = requests.Session()
|
||||||
|
|
@ -32,10 +32,13 @@ class RiceLeafClassificationAPI:
|
||||||
Tuple (success, message)
|
Tuple (success, message)
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
response = self.session.get(f"{self.base_url}/test", timeout=10)
|
response = self.session.get(f"{self.base_url}/health", timeout=10)
|
||||||
if response.status_code == 200:
|
if response.status_code == 200:
|
||||||
data = response.json()
|
data = response.json()
|
||||||
return data['success'], data.get('message', 'Connection OK')
|
if 'status' in data: # Flask API response
|
||||||
|
return data['status'] == 'ok', data.get('message', 'Connection OK')
|
||||||
|
else:
|
||||||
|
return data.get('success', True), data.get('message', 'Connection OK')
|
||||||
else:
|
else:
|
||||||
return False, f"HTTP {response.status_code}"
|
return False, f"HTTP {response.status_code}"
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|
@ -70,7 +73,7 @@ class RiceLeafClassificationAPI:
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# Tentukan endpoint
|
# Tentukan endpoint
|
||||||
endpoint = "classify-and-save" if save else "classify"
|
endpoint = "classify" if not save else "classify"
|
||||||
url = f"{self.base_url}/{endpoint}"
|
url = f"{self.base_url}/{endpoint}"
|
||||||
|
|
||||||
# Baca dan kirim file
|
# Baca dan kirim file
|
||||||
|
|
|
||||||
|
|
@ -1660,7 +1660,7 @@
|
||||||
],
|
],
|
||||||
"metadata": {
|
"metadata": {
|
||||||
"kernelspec": {
|
"kernelspec": {
|
||||||
"display_name": "Python 3 (ipykernel)",
|
"display_name": "base",
|
||||||
"language": "python",
|
"language": "python",
|
||||||
"name": "python3"
|
"name": "python3"
|
||||||
},
|
},
|
||||||
|
|
@ -1674,7 +1674,7 @@
|
||||||
"name": "python",
|
"name": "python",
|
||||||
"nbconvert_exporter": "python",
|
"nbconvert_exporter": "python",
|
||||||
"pygments_lexer": "ipython3",
|
"pygments_lexer": "ipython3",
|
||||||
"version": "3.11.5"
|
"version": "3.12.5"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"nbformat": 4,
|
"nbformat": 4,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue