json is coming back

app.py

@@ -109,19 +109,13 @@ def on_submit(symptoms_text, history):
 
     # Call LLM
     response = llm.complete(prompt)
-    raw = getattr(response, 'text', str(response))
-
-    msg = "📡 Raw LLM response received"
-    log.append(msg)
-    print(msg, flush=True)
-    yield history, None, "\n".join(log)
-
-    # Final assistant message
+    raw = response
     if isinstance(raw, str):
         try:
             raw = json.loads(raw)
         except Exception:
-            pass  # Optionally handle or log the error
+            # If not JSON, wrap in a dict or handle gracefully
+            raw = {"diagnoses": [], "confidences": [], "follow_up": raw}
     assistant_msg = format_response_for_user(raw)
     history = history + [{"role": "assistant", "content": assistant_msg}]
     msg = "✅ Final response appended"