Update app.py

app.py

@@ -261,7 +261,7 @@ def analyze_symptoms_from_text_ai(text_input: str) -> str:
         
         # Extract and clean the response
         try:
-            symptoms_csv = response.choices.message.content.strip()
+            symptoms_csv = response.choices[0].message.content.strip()
             return symptoms_csv
         except Exception as e:
             print ("Error:",e,"\nType 'response': ",type(response),"\n",response) 
@@ -351,7 +351,7 @@ If multiple plants could work well, you may recommend up to 3 options."""
             ]
         )
         
-        final_result = response.choices.message.content
+        final_result = response.choices[0].message.content
         
         # Step 5: Append detailed plant information if available
         if all_related_plants:
@@ -522,6 +522,7 @@ class BotanistAssistant:
             top_p=0.9,
             extra_body={"top_k": 50},
             messages=messages)
+            #return  response.choices[0].message.content #TODO if not working
             return response.to_json()
             
         except Exception as e:
@@ -574,13 +575,12 @@ class BotanistAssistant:
         # Get treatment information
         #treatment_response = search_for_treatment_answer_ai(message)
         #yield treatment_response  # First yield the treatment info
-
+        
         # Stream additional assistant responses if available
         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: