Update main.py

main.py

@@ -89,6 +89,45 @@ def classify_query_with_gemini(query: str):
         print(f"Gemini API call ya JSON parsing mein error: {e}")
         return None
 
+# Helper function to safely call external APIs
+def call_external_api(endpoint_url, category, query, files=None):
+    """External API ko safely call karta hai aur response handle karta hai."""
+    try:
+        # Different categories ke liye different request formats
+        if category == "medicine_info":
+            if files:
+                response = requests.post(endpoint_url, files=files, data={"main_query": query}, timeout=30)
+            else:
+                response = requests.post(endpoint_url, json={"main_query": query}, timeout=30)
+        else:
+            if files:
+                response = requests.post(endpoint_url, files=files, data={"main_query": query}, timeout=30)
+            else:
+                response = requests.post(endpoint_url, data={"main_query": query}, timeout=30)
+        
+        # Status code check karein
+        response.raise_for_status()
+        
+        # Content-Type check karein
+        content_type = response.headers.get('content-type', '')
+        
+        if 'application/json' in content_type:
+            return {"success": True, "data": response.json()}
+        else:
+            # Agar JSON nahi hai to text return karein
+            return {"success": True, "data": {"message": response.text}}
+            
+    except requests.exceptions.ConnectionError:
+        return {"success": False, "error": "API service unavailable. Please check if the service is running."}
+    except requests.exceptions.Timeout:
+        return {"success": False, "error": "API request timed out. Please try again."}
+    except requests.exceptions.HTTPError as e:
+        return {"success": False, "error": f"API returned error: {e.response.status_code}"}
+    except requests.exceptions.JSONDecodeError:
+        return {"success": False, "error": "API returned invalid JSON response"}
+    except Exception as e:
+        return {"success": False, "error": f"Unexpected error: {str(e)}"}
+
 # Step 4: API Routes (Endpoints)
 @app.route('/start_session', methods=['POST'])
 def start_session():
@@ -125,23 +164,36 @@ def process_query():
         }), 200
     else:
         print(f"Session {session_id}: No image required. Forwarding to '{classification.get('category')}' API.")
-        # Asli API ko call karein (Abhi ke liye mock response)
+        
         # Session se query aur classification nikalein
         query = SESSIONS[session_id].get('query')
         classification = SESSIONS[session_id].get('classification')
         category = classification['category']
         endpoint_url = API_ENDPOINTS.get(category)
-        if category == "medicine_info":
-            response = requests.post(endpoint_url, json={"main_query": query})
-        else:
-            response = requests.post(endpoint_url, data={"main_query": query})
+        
+        # External API call with error handling
+        api_result = call_external_api(endpoint_url, category, query)
+        
+        # Session close karein
         del SESSIONS[session_id]
         print(f"Session {session_id} closed.")
-        return jsonify({
-            "status": "success",
-            "response": response.json(),
-            "data": f"Information about '{query}': This is a tuned response from the {classification.get('category')} service."
-        })
+        
+        if api_result["success"]:
+            return jsonify({
+                "status": "success",
+                "response": api_result["data"],
+                "category": category
+            })
+        else:
+            # API call fail ho gaya, mock response bhejein
+            return jsonify({
+                "status": "success",
+                "response": {
+                    "message": f"Service temporarily unavailable. Mock response: Information about '{query}' from {category} service.",
+                    "error_details": api_result["error"]
+                },
+                "category": category
+            })
 
 @app.route('/process_with_image', methods=['POST'])
 def process_with_image():
@@ -165,41 +217,32 @@ def process_with_image():
     
     print(f"Session {session_id}: Image received. Preparing to forward to '{category}' API.")
 
-    # *** NEW: FORWARDING LOGIC THAT MATCHES YOUR CURL COMMAND ***
-    # The file object from Flask needs its stream to be readable by `requests`
-    # We pass the file stream, filename, and mimetype to requests
-    # The dictionary key 'file' matches the '-F file=@...' part of your curl command
+    # File payload prepare karein
     files_payload = {'file': (file.filename, file.stream, file.mimetype)}
     
-    # The dictionary key 'query' matches the '-F query=...' part
-    data_payload = {'query': query}
-
-    try:
-        # NOTE: Neeche di gayi line asli API call hai.
-        # Jab aapka backend service (e.g., http://localhost:5002/api/query) taiyaar ho,
-        # to is line ko uncomment kar dein.
-        
-        response_from_service = requests.post(endpoint_url, files=files_payload, data=data_payload)
-        response_from_service.raise_for_status() # Agar 4xx/5xx error ho to exception raise karega
-        tuned_response = response_from_service.json() # Assume service returns JSON
-
-        # Abhi ke liye, hum ek mock response bhej rahe hain
-        mock_response = {
+    # External API call with error handling
+    api_result = call_external_api(endpoint_url, category, query, files=files_payload)
+    
+    # Session close karein
+    del SESSIONS[session_id]
+    print(f"Session {session_id} closed.")
+    
+    if api_result["success"]:
+        return jsonify({
             "status": "success",
-            "response": tuned_response,
-            "data": f"Analysis for '{query}' based on your image: This is a tuned MOCK response from the {category} service."
-        }
-        
-        # Session close karein
-        del SESSIONS[session_id]
-        print(f"Session {session_id} closed.")
-        
-        return jsonify(mock_response)
-
-    except requests.exceptions.RequestException as e:
-        del SESSIONS[session_id]
-        print(f"Session {session_id} closed after failed API call.")
-        return jsonify({"status": "error", "message": f"Backend service call failed: {e}"}), 503
+            "response": api_result["data"],
+            "category": category
+        })
+    else:
+        # API call fail ho gaya, mock response bhejein
+        return jsonify({
+            "status": "success",
+            "response": {
+                "message": f"Service temporarily unavailable. Mock response: Analysis for '{query}' based on your image from {category} service.",
+                "error_details": api_result["error"]
+            },
+            "category": category
+        })
 
 if __name__ == '__main__':
     app.run(debug=True, port=5000)