Update app.py

app.py

@@ -9,8 +9,7 @@ import gradio as gr
 from dateutil import parser as dateparser
 from fastapi import FastAPI
 from pydantic import BaseModel
-import google.generativeai as genai
-# <<< MISSING IMPORT ADDED BACK >>>
+from groq import AsyncGroq # <<< CHANGED from google.generativeai
 from analytics import record_request, last_n_days_df, last_n_days_avg_time_df
 
 # --- Prompts ---
@@ -72,11 +71,12 @@ async def search_web_logic(query: str, serper_api_key: str, search_type: str, nu
     except Exception as e:
         return f"An error occurred during web search: {str(e)}"
 
-# --- Gemini Summarization Logic ---
-async def summarize_with_gemini(text_to_summarize: str, query: str, gemini_key: str, model_name: str, research_mode: str) -> str:
+# --- Groq Summarization Logic (REPLACED) ---
+async def summarize_with_groq(text_to_summarize: str, query: str, groq_key: str, model_name: str, research_mode: str) -> str:
+    if not groq_key:
+        return "\n\n--- ⚠️ Groq Summarization Skipped ---\nError: API Key is required.\nReturning raw text instead."
     try:
-        genai.configure(api_key=gemini_key)
-        model = genai.GenerativeModel(model_name)
+        client = AsyncGroq(api_key=groq_key)
         current_date = datetime.now(timezone.utc).strftime("%Y-%m-%d")
         
         if research_mode == 'deep':
@@ -85,18 +85,22 @@ async def summarize_with_gemini(text_to_summarize: str, query: str, gemini_key:
             prompt_template = PROMPT_NORMAL
             
         prompt = prompt_template.format(query=query, context_text=text_to_summarize, current_date=current_date)
-        response = await model.generate_content_async(prompt)
-        return response.text
+        
+        chat_completion = await client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
+            model=model_name,
+        )
+        return chat_completion.choices[0].message.content
     except Exception as e:
-        return f"\n\n--- ⚠️ Gemini Summarization Failed ---\nError: {str(e)}\nReturning raw text instead."
+        return f"\n\n--- ⚠️ Groq Summarization Failed ---\nError: {str(e)}\nReturning raw text instead."
 
 # --- Main Orchestrator Function ---
-async def search_and_summarize(query, serper_api_key, search_type, num_results, gemini_api_key, gemini_model, research_mode):
+async def search_and_summarize(query, serper_api_key, search_type, num_results, groq_api_key, groq_model, research_mode):
     scraped_text = await search_web_logic(query, serper_api_key, search_type, num_results)
     
-    if gemini_api_key and "Error:" not in scraped_text:
-        summarized_text = await summarize_with_gemini(scraped_text, query, gemini_api_key, gemini_model, research_mode)
-        if "⚠️ Gemini Summarization Failed" in summarized_text:
+    if groq_api_key and "Error:" not in scraped_text:
+        summarized_text = await summarize_with_groq(scraped_text, query, groq_api_key, groq_model, research_mode) # CHANGED
+        if "⚠️ Groq Summarization Failed" in summarized_text:
              return scraped_text + summarized_text
         else:
              return summarized_text
@@ -104,24 +108,21 @@ async def search_and_summarize(query, serper_api_key, search_type, num_results,
 
 # --- FastAPI App ---
 app = FastAPI()
-# Add CORS middleware if you plan to call the API from a different domain/frontend
-# from fastapi.middleware.cors import CORSMiddleware
-# app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_credentials=True, allow_methods=["*"], allow_headers=["*"])
 
 class SearchRequest(BaseModel):
     query: str
     serper_api_key: str
     search_type: str = "search"
     num_results: int = 4
-    gemini_api_key: Optional[str] = None
-    gemini_model: Optional[str] = "gemini-2.5-flash-lite"
+    groq_api_key: Optional[str] = None # CHANGED
+    groq_model: Optional[str] = "llama3-8b-8192" # CHANGED
     research_mode: str = "normal"
 
 @app.post("/api/search")
 async def api_search(request: SearchRequest):
     result = await search_and_summarize(
         request.query, request.serper_api_key, request.search_type, request.num_results,
-        request.gemini_api_key, request.gemini_model, request.research_mode
+        request.groq_api_key, request.groq_model, request.research_mode
     )
     return {"result": result}
 
@@ -138,16 +139,16 @@ def create_gradio_app():
                     search_type_input = gr.Radio(["search", "news"], value="search", label="Search Type")
                     num_results_input = gr.Slider(1, 20, value=4, step=1, label="Number of Results")
                 
-                gr.Markdown("### Step 2: AI Summarization")
+                gr.Markdown("### Step 2: AI Summarization (Powered by Groq)") # CHANGED
                 research_mode_input = gr.Radio(["normal", "deep"], value="normal", label="Research Mode", info="Normal for fast summary, Deep for detailed report.")
-                gemini_api_key_input = gr.Textbox(label="Your Gemini API Key", type="password", placeholder="Leave empty to skip summarization")
-                gemini_model_input = gr.Textbox(label="Gemini Model", value="gemini-1.5-flash-latest")
+                groq_api_key_input = gr.Textbox(label="Your Groq API Key", type="password", placeholder="Leave empty to skip summarization") # CHANGED
+                groq_model_input = gr.Textbox(label="Groq Model", value="llama3-8b-8192") # CHANGED
                 search_button = gr.Button("Search & Summarize", variant="primary")
                 output = gr.Textbox(label="Result", lines=25, max_lines=40)
                 
                 search_button.click(
                     fn=search_and_summarize,
-                    inputs=[query_input, serper_api_key_input, search_type_input, num_results_input, gemini_api_key_input, gemini_model_input, research_mode_input],
+                    inputs=[query_input, serper_api_key_input, search_type_input, num_results_input, groq_api_key_input, groq_model_input, research_mode_input], # CHANGED
                     outputs=output
                 )
             with gr.Tab("Analytics"):