Update app.py

app.py

@@ -72,26 +72,32 @@ def query_model(model_name: str, messages: List[Dict[str, str]]) -> str:
         return f"{model_name} error: {str(e)}"
 
 def respond(message: str, history: List[List[str]]) -> str:
-    """Handle sequential model responses with progressive context"""
+    """Handle sequential model responses with individual updates"""
     messages = [{"role": "user", "content": message}]
-    responses = []
     
-    # First model: Initial response
-    reply1 = query_model("Qwen2.5-Coder-32B-Instruct", messages)
-    responses.append(f"**Qwen2.5-Coder-32B-Instruct**:\n{reply1}")
-    messages.append({"role": "assistant", "content": reply1})
+    # Get first model's response
+    response1 = query_model("Qwen2.5-Coder-32B-Instruct", messages)
+    yield f"**Qwen2.5-Coder-32B-Instruct**:\n{response1}"
     
-    # Second model: Contextualize with first response
-    reply2 = query_model("Qwen2.5-72B-Instruct", messages)
-    responses.append(f"\n\n**Qwen2.5-72B-Instruct**:\n{reply2}")
-    messages.append({"role": "assistant", "content": reply2})
+    # Add first response to context
+    messages.append({
+        "role": "assistant",
+        "content": f"Previous response: {response1}"
+    })
     
-    # Third model: Final synthesis
-    reply3 = query_model("Llama3.3-70B-Instruct", messages)
-    responses.append(f"\n\n**Llama3.3-70B-Instruct**:\n{reply3}")
+    # Get second model's response
+    response2 = query_model("Qwen2.5-72B-Instruct", messages)
+    yield f"**Qwen2.5-72B-Instruct**:\n{response2}"
     
-    # Return progressive responses with clear separation
-    return "\n\n".join(responses)
+    # Add second response to context
+    messages.append({
+        "role": "assistant",
+        "content": f"Previous responses: {response1}\n{response2}"
+    })
+    
+    # Get final model's response
+    response3 = query_model("Llama3.3-70B-Instruct", messages)
+    yield f"**Llama3.3-70B-Instruct**:\n{response3}"
 
 # Create the Gradio interface
 chat_interface = gr.ChatInterface(