Upload 2 files

app.py

@@ -9,7 +9,8 @@ import gradio as gr
 from dateutil import parser as dateparser
 from fastapi import FastAPI
 from pydantic import BaseModel
-from groq import AsyncGroq # <<< CHANGED from google.generativeai
+import google.generativeai as genai
+# <<< MISSING IMPORT ADDED BACK >>>
 from analytics import record_request, last_n_days_df, last_n_days_avg_time_df
 
 # --- Prompts ---
@@ -71,12 +72,11 @@ 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)}"
 
-# --- 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."
+# --- Gemini Summarization Logic ---
+async def summarize_with_gemini(text_to_summarize: str, query: str, gemini_key: str, model_name: str, research_mode: str) -> str:
     try:
-        client = AsyncGroq(api_key=groq_key)
+        genai.configure(api_key=gemini_key)
+        model = genai.GenerativeModel(model_name)
         current_date = datetime.now(timezone.utc).strftime("%Y-%m-%d")
         
         if research_mode == 'deep':
@@ -85,22 +85,18 @@ async def summarize_with_groq(text_to_summarize: str, query: str, groq_key: str,
             prompt_template = PROMPT_NORMAL
             
         prompt = prompt_template.format(query=query, context_text=text_to_summarize, current_date=current_date)
-        
-        chat_completion = await client.chat.completions.create(
-            messages=[{"role": "user", "content": prompt}],
-            model=model_name,
-        )
-        return chat_completion.choices[0].message.content
+        response = await model.generate_content_async(prompt)
+        return response.text
     except Exception as e:
-        return f"\n\n--- ⚠️ Groq Summarization Failed ---\nError: {str(e)}\nReturning raw text instead."
+        return f"\n\n--- ⚠️ Gemini 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, groq_api_key, groq_model, research_mode):
+async def search_and_summarize(query, serper_api_key, search_type, num_results, gemini_api_key, gemini_model, research_mode):
     scraped_text = await search_web_logic(query, serper_api_key, search_type, num_results)
     
-    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:
+    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:
              return scraped_text + summarized_text
         else:
              return summarized_text
@@ -108,21 +104,24 @@ 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
-    groq_api_key: Optional[str] = None # CHANGED
-    groq_model: Optional[str] = "llama3-8b-8192" # CHANGED
+    gemini_api_key: Optional[str] = None
+    gemini_model: Optional[str] = "gemini-2.5-flash-lite"
     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.groq_api_key, request.groq_model, request.research_mode
+        request.gemini_api_key, request.gemini_model, request.research_mode
     )
     return {"result": result}
 
@@ -139,16 +138,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 (Powered by Groq)") # CHANGED
+                gr.Markdown("### Step 2: AI Summarization")
                 research_mode_input = gr.Radio(["normal", "deep"], value="normal", label="Research Mode", info="Normal for fast summary, Deep for detailed report.")
-                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
+                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")
                 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, groq_api_key_input, groq_model_input, research_mode_input], # CHANGED
+                    inputs=[query_input, serper_api_key_input, search_type_input, num_results_input, gemini_api_key_input, gemini_model_input, research_mode_input],
                     outputs=output
                 )
             with gr.Tab("Analytics"):

requirements.txt

@@ -1,3 +1,5 @@
+# requirements.txt
+
 gradio
 httpx
 trafilatura
@@ -6,4 +8,4 @@ limits
 filelock
 fastapi
 uvicorn
-groq
+google-generativeai