Update main.py

main.py

@@ -93,17 +93,70 @@ def classify_query_with_gemini(query: str):
 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
+        # Category-specific request handling based on your actual APIs
         if category == "medicine_info":
+            # Medicine API expects JSON format with "query" parameter
             if files:
-                response = requests.post(endpoint_url, files=files, data={"query": query}, timeout=30)
+                # For file upload, use multipart form data
+                response = requests.post(
+                    endpoint_url, 
+                    files=files,
+                    data={"query": query},  # Medicine API uses "query" not "main_query"
+                    timeout=30
+                )
             else:
-                response = requests.post(endpoint_url, json={"query": query}, timeout=30)
+                # For text-only queries, use JSON
+                response = requests.post(
+                    endpoint_url, 
+                    json={"query": query},  # Medicine API uses "query" not "main_query"
+                    headers={'Content-Type': 'application/json'},
+                    timeout=30
+                )
+        
+        elif category == "disease_query":
+            # Disease query API - try multiple formats since we're not sure
+            if files:
+                response = requests.post(
+                    endpoint_url, 
+                    files=files, 
+                    data={"main_query": query},
+                    timeout=30
+                )
+            else:
+                # Try JSON first, fallback to form data if needed
+                try:
+                    response = requests.post(
+                        endpoint_url, 
+                        json={"main_query": query}, 
+                        headers={'Content-Type': 'application/json'},
+                        timeout=30
+                    )
+                except requests.exceptions.HTTPError as json_error:
+                    if json_error.response.status_code == 415:
+                        # Try form data format
+                        response = requests.post(
+                            endpoint_url, 
+                            data={"main_query": query}, 
+                            timeout=30
+                        )
+                    else:
+                        raise
+        
         else:
+            # For other categories (skin_disease, report_reading)
             if files:
-                response = requests.post(endpoint_url, files=files, data={"query": query},json={"query": query}, timeout=30)
+                response = requests.post(
+                    endpoint_url, 
+                    files=files, 
+                    data={"main_query": query}, 
+                    timeout=30
+                )
             else:
-                response = requests.post(endpoint_url,data={"query": query}, json={"query": query}, timeout=30)
+                response = requests.post(
+                    endpoint_url, 
+                    data={"main_query": query}, 
+                    timeout=30
+                )
         
         # Status code check karein
         response.raise_for_status()
@@ -136,6 +189,65 @@ def start_session():
     print(f"Session started: {session_id}")
     return jsonify({"session_id": session_id}), 200
 
+# Temporary debugging endpoint
+@app.route('/test_api', methods=['POST'])
+def test_api():
+    """Debugging endpoint to test different API formats"""
+    data = request.get_json()
+    query = data.get('query', 'test query')
+    endpoint_url = data.get('endpoint', 'http://localhost:5001/ask')
+    
+    results = {}
+    
+    # Test 1: JSON format
+    try:
+        response = requests.post(
+            endpoint_url, 
+            json={"main_query": query}, 
+            headers={'Content-Type': 'application/json'},
+            timeout=10
+        )
+        results['json_format'] = {
+            'status_code': response.status_code,
+            'content_type': response.headers.get('content-type'),
+            'response': response.text[:200] if response.text else 'Empty'
+        }
+    except Exception as e:
+        results['json_format'] = {'error': str(e)}
+    
+    # Test 2: Form data format
+    try:
+        response = requests.post(
+            endpoint_url, 
+            data={"main_query": query}, 
+            timeout=10
+        )
+        results['form_format'] = {
+            'status_code': response.status_code,
+            'content_type': response.headers.get('content-type'),
+            'response': response.text[:200] if response.text else 'Empty'
+        }
+    except Exception as e:
+        results['form_format'] = {'error': str(e)}
+    
+    # Test 3: Alternative parameter name
+    try:
+        response = requests.post(
+            endpoint_url, 
+            json={"query": query}, 
+            headers={'Content-Type': 'application/json'},
+            timeout=10
+        )
+        results['alt_param_json'] = {
+            'status_code': response.status_code,
+            'content_type': response.headers.get('content-type'),
+            'response': response.text[:200] if response.text else 'Empty'
+        }
+    except Exception as e:
+        results['alt_param_json'] = {'error': str(e)}
+    
+    return jsonify(results)
+
 @app.route('/process_query', methods=['POST'])
 def process_query():
     data = request.get_json()
@@ -217,8 +329,13 @@ def process_with_image():
     
     print(f"Session {session_id}: Image received. Preparing to forward to '{category}' API.")
 
-    # File payload prepare karein
-    files_payload = {'file': (file.filename, file.stream, file.mimetype)}
+    # File payload prepare karein - different for different categories
+    if category == "medicine_info":
+        # Medicine API expects 'file' as the key
+        files_payload = {'file': (file.filename, file.stream, file.mimetype)}
+    else:
+        # Other APIs might expect 'photo' or 'image'
+        files_payload = {'file': (file.filename, file.stream, file.mimetype)}
     
     # External API call with error handling
     api_result = call_external_api(endpoint_url, category, query, files=files_payload)
@@ -243,6 +360,3 @@ def process_with_image():
             },
             "category": category
         })
-
-if __name__ == '__main__':
-    app.run(debug=True, port=5000)