Update app.py

app.py

@@ -4,41 +4,53 @@ from google import genai
 from google.genai.types import GenerateContentConfig, GoogleSearch, Tool
 
 # Initialize GenAI Client
-API_KEY = os.getenv("GOOGLE_API_KEY")  # Ensure to set this in Hugging Face Secrets
+API_KEY = os.getenv("GOOGLE_API_KEY")  # Hugging Face Secrets
 client = genai.Client(api_key=API_KEY)
-MODEL_ID = "gemini-2.0-flash"  # Replace with your desired model ID
+MODEL_ID = "gemini-2.0-flash"
 
-def google_search_query(question):
+def fact_check_claim(claim):
     try:
-        # Define the Google Search Tool
+        # Google Search Tool
         google_search_tool = Tool(google_search=GoogleSearch())
-        
-        # Generate the response
+
+        prompt = f"""
+You are a fact-checking assistant.
+Analyze the following claim and decide whether it is:
+- True
+- False
+- Misleading
+
+Then give a short explanation based on reliable sources.
+
+Claim:
+{claim}
+"""
+
         response = client.models.generate_content(
             model=MODEL_ID,
-            contents=question,
+            contents=prompt,
             config=GenerateContentConfig(tools=[google_search_tool]),
         )
 
-        # Extract AI response and search results
-        ai_response = response.text  # AI response as plain text
-        search_results = response.candidates[0].grounding_metadata.search_entry_point.rendered_content
+        ai_analysis = response.text
+        sources_html = response.candidates[0].grounding_metadata.search_entry_point.rendered_content
+
+        return ai_analysis, sources_html
 
-        return ai_response, search_results
     except Exception as e:
         return f"Error: {str(e)}", ""
 
 # Gradio Interface
 app = gr.Interface(
-    fn=google_search_query,
-    inputs=gr.Textbox(lines=2, label="Ask a Question"),
+    fn=fact_check_claim,
+    inputs=gr.Textbox(lines=3, label="Enter a claim or news statement"),
     outputs=[
-        gr.Textbox(label="AI Response"),
-        gr.HTML(label="Search Results"),
+        gr.Textbox(label="AI Fact Check Result"),
+        gr.HTML(label="Sources & Evidence"),
     ],
-    title="Google Search with Gemini AI",
-    description="Ask a question, and the AI will use Google search tools to provide an answer along with contextual search results.",
+    title="AI Fact Checker with Gemini",
+    description="Enter a claim or news statement. The AI will verify it using Google search and provide evidence-based analysis.",
 )
 
 if __name__ == "__main__":
-    app.launch(share=True)
+    app.launch(share=True)