Update app.py

app.py

@@ -1,7 +1,7 @@
 from typing import List, Tuple, Dict, Any, Generator
 import sqlite3 
 import urllib
-#import requests
+import requests
 import os
 import gradio as gr
 import time
@@ -260,8 +260,12 @@ def analyze_symptoms_from_text_ai(text_input: str) -> str:
         )
         
         # Extract and clean the response
-        symptoms_csv = response.choices.message.content.strip()
-        return symptoms_csv
+        try:
+            symptoms_csv = response.choices.message.content.strip()
+            return symptoms_csv
+        except Exception as e:
+            print ("Error:",e,"\nType 'response': ",type(response),"\n",response) 
+        
         
     except Exception as e:
         print(f"Error analyzing symptoms: {str(e)}")
@@ -518,7 +522,7 @@ class BotanistAssistant:
             top_p=0.9,
             extra_body={"top_k": 50},
             messages=messages)
-            return response.choices.message.content
+            return response.to_json()
             
         except Exception as e:
             print(f"Error: {str(e)}")
@@ -572,8 +576,12 @@ class BotanistAssistant:
         #yield treatment_response  # First yield the treatment info
 
         # Stream additional assistant responses if available
-        if api_response and api_response.choices:
-            yield api_response.choices.message.content
+        if "choices" in api_response:
+            for choice in api_response["choices"]:
+                if "message" in choice and "content" in choice["message"]:
+                    yield choice["message"]["content"]
+        
+                
 
 def create_app(api_endpoint: str = "https://api.studio.nebius.com/v1/") -> gr.Blocks:
     """Creates and configures the Gradio interface."""