modular code apsssjjj

core.py

@@ -4,7 +4,7 @@ from typing import AsyncGenerator, List, Dict, Tuple
 from config import logger
 from api import ask_openai, ask_anthropic, ask_gemini
 
-async def query_model(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]]], None]:
+async def query_model(query: str, providers: List[str], history: List[Dict[str, str]]) -> AsyncGenerator[Tuple[str, str, List[Dict[str, str]], List[Dict[str, str]], List[Dict[str, str]]], None]:
     logger.info(f"Processing query with providers: {providers}")
     openai_response = ""
     anthropic_response = ""
@@ -92,7 +92,7 @@ async def query_model(query: str, providers: List[str], history: List[Dict[str,
                         gemini_response = response
                         gemini_messages = messages
                     logger.info(f"Yielding update for {provider}: {response[:50]}...")
-                    yield "", openai_messages, anthropic_messages, gemini_messages
+                    yield response, openai_messages, anthropic_messages, gemini_messages
                     new_generator_states[new_generator_states.index((provider, gen, task))] = (provider, gen, None)
                 except StopAsyncIteration:
                     logger.info(f"Generator for {provider} completed")
@@ -103,19 +103,18 @@ async def query_model(query: str, providers: List[str], history: List[Dict[str,
 
         active_generators = new_generator_states
 
-    responses = []
+    combined_response = ""
     if openai_response.strip() and not openai_response.startswith("Error:"):
-        responses.append(f"[OpenAI]: {openai_response}")
+        combined_response += f"[OpenAI]: {openai_response}\n\n"
     if anthropic_response.strip() and not anthropic_response.startswith("Error:"):
-        responses.append(f"[Anthropic]: {anthropic_response}")
+        combined_response += f"[Anthropic]: {anthropic_response}\n\n"
     if gemini_response.strip() and not gemini_response.startswith("Error:"):
-        responses.append(f"[Gemini]: {gemini_response}")
+        combined_response += f"[Gemini]: {gemini_response}"
 
-    combined_response = "\n\n".join(responses) if responses else "No valid responses received."
-    updated_history = history + [{"user": query, "bot": combined_response}]
+    updated_history = history + [{"user": query, "bot": combined_response.strip()}]
     logger.info(f"Updated history: {updated_history}")
 
-    yield combined_response, openai_messages, anthropic_messages, gemini_messages
+    yield combined_response.strip(), 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]: