Final Fix 😅

app.py

@@ -522,8 +522,8 @@ 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()
+            return  response.choices[0].message.content #TODO if not working
+            #return response.to_json()
             
         except Exception as e:
             print(f"Error: {str(e)}")
@@ -577,11 +577,14 @@ class BotanistAssistant:
         #yield treatment_response  # First yield the treatment info
         
         # Stream additional assistant responses if available
+        if api_response:
+            yield api_response
+        '''
         if "choices" in api_response:
             for choice in api_response["choices"][0]:
                 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."""