see a progress counter when submit is it

src/app.py

@@ -555,25 +555,31 @@ with gr.Blocks(theme="default") as demo:
             return history, ""  # Return tuple to clear input
 
         try:
-            # Process the symptoms
-            diagnosis_query = f"""
-            Based on these symptoms: '{text}'
-            Provide relevant ICD-10 codes and diagnostic questions.
-            Focus on clinical implications.
-            Limit response to 1000 characters.
-            """
-            response = symptom_index.as_query_engine().query(diagnosis_query)
+            # Process the symptoms using the configured LLM
+            prompt = f"""Given these symptoms: '{text}'
+            Please provide:
+            1. Most likely ICD-10 codes
+            2. Confidence levels for each diagnosis
+            3. Key follow-up questions
+
+            Format as JSON with diagnoses, confidences, and follow_up fields."""
             
-            # Clean up memory
-            cleanup_memory()
+            response = llm.complete(prompt)
+            
+            try:
+                # Try to parse as JSON first
+                result = json.loads(response.text)
+            except json.JSONDecodeError:
+                # If not JSON, wrap in our format
+                result = {
+                    "diagnoses": [],
+                    "confidences": [],
+                    "follow_up": str(response.text)[:1000]  # Limit response length
+                }
 
             new_history = history + [
                 {"role": "user", "content": text},
-                {"role": "assistant", "content": format_response_for_user({
-                    "diagnoses": [],
-                    "confidences": [],
-                    "follow_up": str(response)[:1000]
-                })}
+                {"role": "assistant", "content": format_response_for_user(result)}
             ]
             return new_history, ""  # Return empty string to clear input
         except Exception as e:
@@ -584,9 +590,9 @@ with gr.Blocks(theme="default") as demo:
     submit_btn.click(
         fn=process_text_input,
         inputs=[text_input, chatbot],
-        outputs=[chatbot, text_input],  # Add text_input to outputs
+        outputs=[chatbot, text_input],
         queue=True
-    ).then(  # Chain cleanup
+    ).success(  # Changed from .then to .success for better error handling
         fn=cleanup_memory,
         inputs=None,
         outputs=None,