modular code apsssjjjadfdr

app.py

@@ -1,13 +1,12 @@
-
 import gradio as gr
 from core import submit_query, clear_history
 
-with gr.Blocks(theme=gr.themes.Soft(), css="""
+with gr.Blocks(theme=gr.themes.Soft(), css=\"\"\"
 .full-height { height: 100%; display: flex; align-items: stretch; min-height: 40px; }
 .full-height button { height: 100%; padding: 8px 16px; }
 .providers-row { height: 100%; display: flex; align-items: stretch; min-height: 40px; }
 .providers-row .checkbox-group { height: 100%; display: flex; flex-direction: row; align-items: center; gap: 10px; }
-""") as demo:
+\"\"\") as demo:
     gr.Markdown("# Multi-Model Chat")
     gr.Markdown("Chat with OpenAI, Anthropic, or Gemini. Select providers and compare responses side by side!")
 

core.py

@@ -1,4 +1,3 @@
-
 import asyncio
 from typing import AsyncGenerator, List, Dict, Tuple
 from config import logger
@@ -18,12 +17,15 @@ async def query_model(query: str, providers: List[str], history: List[Dict[str,
         openai_messages.append({"role": "user", "content": msg["user"]})
         anthropic_messages.append({"role": "user", "content": msg["user"]})
         gemini_messages.append({"role": "user", "content": msg["user"]})
-        if "openai" in msg and msg["openai"]:
-            openai_messages.append({"role": "assistant", "content": msg["openai"]})
-        if "anthropic" in msg and msg["anthropic"]:
-            anthropic_messages.append({"role": "assistant", "content": msg["anthropic"]})
-        if "gemini" in msg and msg["gemini"]:
-            gemini_messages.append({"role": "assistant", "content": msg["gemini"]})
+        if msg["bot"]:
+            lines = msg["bot"].split("\\n\\n")
+            for line in lines:
+                if line.startswith("[OpenAI]:"):
+                    openai_messages.append({"role": "assistant", "content": line[len("[OpenAI]:"):].strip()})
+                elif line.startswith("[Anthropic]:"):
+                    anthropic_messages.append({"role": "assistant", "content": line[len("[Anthropic]:"):].strip()})
+                elif line.startswith("[Gemini]:"):
+                    gemini_messages.append({"role": "assistant", "content": line[len("[Gemini]:"):].strip()})
 
     if "OpenAI" in providers:
         openai_messages.append({"role": "user", "content": query})
@@ -100,19 +102,22 @@ async def query_model(query: str, providers: List[str], history: List[Dict[str,
 
         active_generators = new_generator_states
 
-    updated_entry = {
-        "user": query,
-        "openai": openai_response if openai_response.strip() and not openai_response.startswith("Error:") else "",
-        "anthropic": anthropic_response if anthropic_response.strip() and not anthropic_response.startswith("Error:") else "",
-        "gemini": gemini_response if gemini_response.strip() and not gemini_response.startswith("Error:") else ""
-    }
+    responses = []
+    if openai_response.strip() and not openai_response.startswith("Error:"):
+        responses.append(f"[OpenAI]: {openai_response}")
+    if anthropic_response.strip() and not anthropic_response.startswith("Error:"):
+        responses.append(f"[Anthropic]: {anthropic_response}")
+    if gemini_response.strip() and not gemini_response.startswith("Error:"):
+        responses.append(f"[Gemini]: {gemini_response}")
 
-    updated_history = history + [updated_entry]
-    logger.info(f"Updated structured history: {updated_history}")
+    combined_response = "\\n\\n".join(responses) if responses else "No valid responses received."
+    updated_history = history + [{"user": query, "bot": combined_response}]
+    logger.info(f"Updated history: {updated_history}")
 
-    yield "", openai_messages, anthropic_messages, gemini_messages
+    yield combined_response, openai_messages, anthropic_messages, gemini_messages
 
-async def submit_query(query: str, providers: List[str], history: List[Dict[str, str]])     -> AsyncGenerator[Tuple[str, List[Dict[str, str]], List[Dict[str, str]], List[Dict[str, str]], List[Dict[str, str]]], None]:
+async def submit_query(query: str, providers: List[str], history: List[Dict[str, str]]) \
+    -> AsyncGenerator[Tuple[str, List[Dict[str, str]], List[Dict[str, str]], List[Dict[str, str]], List[Dict[str, str]]], None]:
 
     if not query.strip():
         msg = {"role": "assistant", "content": "Please enter a query."}
@@ -125,13 +130,8 @@ async def submit_query(query: str, providers: List[str], history: List[Dict[str,
         return
 
     final_history = history.copy()
-    async for _, openai_messages, anthropic_messages, gemini_messages in query_model(query, providers, history):
-        final_history = history + [{
-            "user": query,
-            "openai": openai_messages[-1]["content"] if openai_messages else "",
-            "anthropic": anthropic_messages[-1]["content"] if anthropic_messages else "",
-            "gemini": gemini_messages[-1]["content"] if gemini_messages else ""
-        }]
+    async for response_chunk, openai_messages, anthropic_messages, gemini_messages in query_model(query, providers, history):
+        final_history = history + [{"user": query, "bot": response_chunk}]
         yield "", openai_messages, anthropic_messages, gemini_messages, final_history
 
     yield "", openai_messages, anthropic_messages, gemini_messages, final_history