Update app.py

app.py

@@ -78,14 +78,6 @@ st.markdown("""
         box-shadow: 0 2px 8px rgba(0,0,0,0.1);
         color: #856404 !important;
     }
-    /* Section titles (bold black headings) */
-    .section-title {
-        font-size: 1.7rem;
-        font-weight: 800;
-        color: #000000 !important;
-        margin-top: 2rem;
-        margin-bottom: 1rem;
-    }
     
     .verdict-mixed h3, .verdict-mixed p, .verdict-mixed strong {
         color: #856404 !important;
@@ -123,6 +115,7 @@ st.markdown("""
         border: 2px solid #dee2e6;
         box-shadow: 0 4px 6px rgba(0,0,0,0.1);
     }
+    
     /* Fix input & textarea visibility */
     textarea, input {
         background-color: #f8f9fa !important;
@@ -135,6 +128,7 @@ st.markdown("""
     textarea::placeholder, input::placeholder {
         color: #6c757d !important;
     }
+
     .stat-box h2 {
         margin: 0 !important;
         font-size: 2.5rem !important;
@@ -148,43 +142,33 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
 
-
+# ==============================
+# Google Fact Check API Class
+# ==============================
 class GoogleFactCheckAPI:
-    """Google Fact Check Tools API Integration - IMPROVED VERSION"""
+    """Google Fact Check Tools API Integration"""
     
     def __init__(self, api_key: Optional[str] = None):
         self.api_key = api_key or os.environ.get('GOOGLE_API_KEY')
         self.base_url = "https://factchecktools.googleapis.com/v1alpha1/claims:search"
-        
         if not self.api_key:
-            st.warning("⚠️ Google API Key not found. Add it in Settings → Repository secrets")
+            st.markdown("""
+            <div class="verdict-mixed">
+                <p>⚠️ Google API Key not found. Add it in Settings → Repository secrets</p>
+            </div>
+            """, unsafe_allow_html=True)
     
     def verify_claim(self, claim: str, language: str = "en") -> Dict:
-        """
-        Verify claim with IMPROVED search strategy
-        """
         if not self.api_key:
-            return {
-                "status": "error",
-                "message": "❌ API key not configured. Please add GOOGLE_API_KEY in Hugging Face Secrets."
-            }
+            return {"status": "error", "message": "❌ API key not configured. Please add GOOGLE_API_KEY in Hugging Face Secrets."}
         
         try:
-            # Try original query first
-            params = {
-                "key": self.api_key,
-                "query": claim,
-                "languageCode": language,
-                "pageSize": 10  # Get more results
-            }
-            
+            params = {"key": self.api_key, "query": claim, "languageCode": language, "pageSize": 10}
             response = requests.get(self.base_url, params=params, timeout=15)
             response.raise_for_status()
             data = response.json()
             
-            # If no results, try simplified query
             if "claims" not in data or len(data["claims"]) == 0:
-                # Try extracting key terms
                 simplified_query = self._simplify_claim(claim)
                 if simplified_query != claim:
                     params["query"] = simplified_query
@@ -193,17 +177,11 @@ class GoogleFactCheckAPI:
                     data = response.json()
             
             if "claims" not in data or len(data["claims"]) == 0:
-                return {
-                    "status": "no_results",
-                    "message": f"No fact-checks found for: '{claim}'"
-                }
+                return {"status": "no_results", "message": f"No fact-checks found for: '{claim}'"}
             
-            # Process results
             results = []
             for claim_data in data["claims"]:
-                claim_review = claim_data.get("claimReview", [])
-                
-                for review in claim_review:
+                for review in claim_data.get("claimReview", []):
                     results.append({
                         "claim_text": claim_data.get("text", ""),
                         "claimant": claim_data.get("claimant", "Unknown"),
@@ -215,124 +193,85 @@ class GoogleFactCheckAPI:
                         "language": review.get("languageCode", "en")
                     })
             
-            return {
-                "status": "success",
-                "claim": claim,
-                "results": results,
-                "count": len(results)
-            }
-            
+            return {"status": "success", "claim": claim, "results": results, "count": len(results)}
+        
         except requests.exceptions.HTTPError as e:
             if e.response.status_code == 403:
-                return {
-                    "status": "error",
-                    "message": "❌ API key is invalid or Fact Check API is not enabled. Check Google Cloud Console."
-                }
-            return {
-                "status": "error",
-                "message": f"❌ HTTP Error: {str(e)}"
-            }
+                return {"status": "error", "message": "❌ API key is invalid or Fact Check API is not enabled. Check Google Cloud Console."}
+            return {"status": "error", "message": f"❌ HTTP Error: {str(e)}"}
         except requests.exceptions.RequestException as e:
-            return {
-                "status": "error",
-                "message": f"❌ Network error: {str(e)}"
-            }
+            return {"status": "error", "message": f"❌ Network error: {str(e)}"}
         except Exception as e:
-            return {
-                "status": "error",
-                "message": f"❌ Unexpected error: {str(e)}"
-            }
+            return {"status": "error", "message": f"❌ Unexpected error: {str(e)}"}
     
     def _simplify_claim(self, claim: str) -> str:
-        """Extract key terms from claim for better matching"""
-        # Remove common words that don't help matching
-        stop_words = ['the', 'a', 'an', 'is', 'are', 'was', 'were', 'be', 'been', 
-                     'being', 'have', 'has', 'had', 'do', 'does', 'did', 'will', 
-                     'would', 'should', 'could', 'may', 'might', 'can']
-        
+        stop_words = ['the', 'a', 'an', 'is', 'are', 'was', 'were', 'be', 'been', 'being',
+                      'have', 'has', 'had', 'do', 'does', 'did', 'will', 'would', 'should',
+                      'could', 'may', 'might', 'can']
         words = claim.lower().split()
         key_words = [w for w in words if w not in stop_words and len(w) > 2]
-        
-        return ' '.join(key_words[:5]) if key_words else claim  # Max 5 key words
-
+        return ' '.join(key_words[:5]) if key_words else claim
 
+# ==============================
+# Helper Functions
+# ==============================
 def determine_verdict_color(rating: str) -> str:
-    """Determine CSS class based on rating"""
     rating_lower = rating.lower()
-    
     true_keywords = ["true", "correct", "accurate", "mostly true", "verified", "confirmed"]
     false_keywords = ["false", "incorrect", "inaccurate", "mostly false", "debunked", "fake", "pants on fire"]
     mixed_keywords = ["mixture", "mixed", "partially", "misleading", "unproven", "undetermined"]
-    
-    if any(keyword in rating_lower for keyword in true_keywords):
+    if any(k in rating_lower for k in true_keywords):
         return "verdict-true"
-    elif any(keyword in rating_lower for keyword in false_keywords):
+    elif any(k in rating_lower for k in false_keywords):
         return "verdict-false"
-    elif any(keyword in rating_lower for keyword in mixed_keywords):
+    elif any(k in rating_lower for k in mixed_keywords):
         return "verdict-mixed"
-    else:
-        return "verdict-unverified"
-
+    return "verdict-unverified"
 
 def get_verdict_emoji(rating: str) -> str:
-    """Get emoji based on verdict"""
-    verdict_class = determine_verdict_color(rating)
-    
-    emoji_map = {
-        "verdict-true": "✅",
-        "verdict-false": "❌",
-        "verdict-mixed": "⚠️",
-        "verdict-unverified": "❓"
-    }
-    
-    return emoji_map.get(verdict_class, "❓")
+    return {"verdict-true":"✅","verdict-false":"❌","verdict-mixed":"⚠️","verdict-unverified":"❓"}.get(determine_verdict_color(rating),"❓")
 
 def classify_rating(rating: str) -> str:
-    """Classify rating into true/false/mixed/unverified"""
     r = rating.lower()
-
-    if any(k in r for k in ["false", "fake", "pants on fire"]):
-        return "false"
-    elif any(k in r for k in ["true", "correct", "accurate"]):
-        return "true"
-    elif any(k in r for k in ["mixed", "misleading", "partially", "half", "mostly"]):
-        return "mixed"
+    if any(k in r for k in ["false","fake","pants on fire"]): return "false"
+    if any(k in r for k in ["true","correct","accurate"]): return "true"
+    if any(k in r for k in ["mixed","misleading","partially","half","mostly"]): return "mixed"
     return "unverified"
 
+# ==============================
+# Main App
+# ==============================
 def main():
-    # Header with FIXED colors
     st.markdown('<div class="main-header">🔍 VeriFact</div>', unsafe_allow_html=True)
     st.markdown('<div class="sub-header">AI-Powered Fact Checking • Built by Areeba Fatima</div>', unsafe_allow_html=True)
-    
+
     # Sidebar
     with st.sidebar:
         st.header("⚙️ Configuration")
-        
-        api_key_input = st.text_input(
-            "Google API Key (Optional)",
-            type="password",
-            help="Leave empty to use Hugging Face Secrets"
-        )
-        
+        api_key_input = st.text_input("Google API Key (Optional)", type="password", help="Leave empty to use Hugging Face Secrets")
         st.markdown("---")
         
-        st.success("""
-        **VeriFact** verifies claims using Google's Fact Check Tools API against 
-        databases from PolitiFact, Snopes, FactCheck.org, and 50+ other sources.
-        """)
+        st.markdown("""
+        <div class="verdict-unverified">
+            <p>ℹ️ <strong>VeriFact</strong> verifies claims using Google's Fact Check Tools API against
+            databases from PolitiFact, Snopes, FactCheck.org, and 50+ other sources.</p>
+        </div>
+        """, unsafe_allow_html=True)
         
         st.markdown("---")
-        
-        st.info("""
-        **How to Use:**
-        1. Enter any claim
-        2. Click Verify
-        3. Review results
-        4. Check sources
-        """)
+        st.markdown("""
+        <div class="verdict-unverified">
+            <p>ℹ️ <strong>How to Use:</strong></p>
+            <ul>
+                <li>Enter any claim</li>
+                <li>Click Verify</li>
+                <li>Review results</li>
+                <li>Check sources</li>
+            </ul>
+        </div>
+        """, unsafe_allow_html=True)
         
         st.markdown("---")
-        
         with st.expander("📝 Example Claims to Try"):
             st.code("""
 ✅ TRUE:
@@ -355,33 +294,22 @@ def main():
     fact_checker = GoogleFactCheckAPI(api_key)
     
     # Main content
-    st.markdown(
-        '<div class="section-title">📝 Enter Your Claim</div>',
-        unsafe_allow_html=True
-    )
+    st.markdown('<div class="section-title">📝 Enter Your Claim</div>', unsafe_allow_html=True)
+    claim_text = st.text_area("Claim to Verify", height=120, placeholder="Example: COVID-19 vaccines are safe and effective", help="Enter any factual claim you want to verify")
     
-    claim_text = st.text_area(
-        "Claim to Verify",
-        height=120,
-        placeholder="Example: COVID-19 vaccines are safe and effective",
-        help="Enter any factual claim you want to verify"
-    )
-    
-    col1, col2, col3 = st.columns([2, 2, 3])
-    
-    with col1:
-        verify_button = st.button("🔍 Verify Claim", type="primary", use_container_width=True)
-    
-    with col2:
-        clear_button = st.button("🗑️ Clear", use_container_width=True)
-    
-    if clear_button:
-        st.rerun()
+    col1, col2, col3 = st.columns([2,2,3])
+    with col1: verify_button = st.button("🔍 Verify Claim", type="primary", use_container_width=True)
+    with col2: clear_button = st.button("🗑️ Clear", use_container_width=True)
+    if clear_button: st.rerun()
     
     # Process verification
     if verify_button:
         if not claim_text.strip():
-            st.warning("⚠️ Please enter a claim to verify.")
+            st.markdown("""
+            <div class="verdict-mixed">
+                <p>⚠️ Please enter a claim to verify.</p>
+            </div>
+            """, unsafe_allow_html=True)
         else:
             with st.spinner("🔎 Searching fact-check databases..."):
                 start_time = time.time()
@@ -391,95 +319,73 @@ def main():
             st.markdown("---")
             
             if results["status"] == "error":
-                st.error(results['message'])
-                st.info("""
-                **Troubleshooting:**
-                1. Make sure you added `GOOGLE_API_KEY` in Settings → Repository secrets
-                2. Verify the API key is correct
-                3. Ensure Fact Check Tools API is enabled in Google Cloud Console
-                """)
-                
+                st.markdown(f"""
+                <div class="verdict-false">
+                    <p>❌ {results['message']}</p>
+                </div>
+                <div class="verdict-unverified">
+                    <p>ℹ️ <strong>Troubleshooting:</strong></p>
+                    <ul>
+                        <li>Make sure you added <code>GOOGLE_API_KEY</code> in Settings → Repository secrets</li>
+                        <li>Verify the API key is correct</li>
+                        <li>Ensure Fact Check Tools API is enabled in Google Cloud Console</li>
+                    </ul>
+                </div>
+                """, unsafe_allow_html=True)
+            
             elif results["status"] == "no_results":
-                st.warning(results['message'])
-                st.info("""
-                **Why no results?**
-                • This specific claim may not have been fact-checked yet
-                • Try rephrasing (simpler is better)
-                • Try well-known claims like "The Earth is flat"
-                • Check spelling
-                
-                **Note:** Not all claims have fact-checks available. The database contains 
-                claims that have been verified by major fact-checking organizations.
-                """)
-                
+                st.markdown(f"""
+                <div class="verdict-mixed">
+                    <p>⚠️ {results['message']}</p>
+                </div>
+                <div class="verdict-unverified">
+                    <p>ℹ️ <strong>Why no results?</strong></p>
+                    <ul>
+                        <li>This specific claim may not have been fact-checked yet</li>
+                        <li>Try rephrasing (simpler is better)</li>
+                        <li>Try well-known claims like "The Earth is flat"</li>
+                        <li>Check spelling</li>
+                    </ul>
+                    <p><strong>Note:</strong> Not all claims have fact-checks available. The database contains 
+                    claims that have been verified by major fact-checking organizations.</p>
+                </div>
+                """, unsafe_allow_html=True)
+            
             elif results["status"] == "success":
-                st.success(f"✅ Found {results['count']} fact-check(s) in {end_time - start_time:.2f}s")
+                st.markdown(f"""
+                <div class="verdict-true">
+                    <h3>✅ Found {results['count']} fact-check(s)</h3>
+                    <p>⏱ Time taken: {end_time - start_time:.2f}s</p>
+                </div>
+                """, unsafe_allow_html=True)
+                
+                st.markdown('<div class="section-title">📊 Results</div>', unsafe_allow_html=True)
+                st.markdown("""
+                <div class="verdict-unverified">
+                    <p>ℹ️ <strong>Important:</strong> VeriFact does not perform original fact-checking. 
+                    It retrieves claims that have already been verified by professional, 
+                    published fact-checking organizations approved by Google.</p>
+                </div>
+                """, unsafe_allow_html=True)
                 
-                # Display statistics
-                st.markdown(
-                    '<div class="section-title">📊 Results</div>',
-                    unsafe_allow_html=True
-                )
-
-                st.info(
-                    "ℹ️ **Important:** VeriFact does not perform original fact-checking. "
-                    "It retrieves claims that have already been verified by professional, "
-                    "published fact-checking organizations approved by Google."
-                )
-                # Count ratings
                 ratings = [r["rating"] for r in results["results"]]
-
                 classified = [classify_rating(r) for r in ratings]
-
                 true_count = classified.count("true")
                 false_count = classified.count("false")
                 mixed_count = classified.count("mixed")
-
-                col1, col2, col3, col4 = st.columns(4)
-                
-                with col1:
-                    st.markdown(f"""
-                    <div class="stat-box">
-                        <h2 style="color: #28a745;">✅ {true_count}</h2>
-                        <p>True/Accurate</p>
-                    </div>
-                    """, unsafe_allow_html=True)
-                
-                with col2:
-                    st.markdown(f"""
-                    <div class="stat-box">
-                        <h2 style="color: #dc3545;">❌ {false_count}</h2>
-                        <p>False/Incorrect</p>
-                    </div>
-                    """, unsafe_allow_html=True)
                 
-                with col3:
-                    st.markdown(f"""
-                    <div class="stat-box">
-                        <h2 style="color: #ffc107;">⚠️ {mixed_count}</h2>
-                        <p>Mixed/Misleading</p>
-                    </div>
-                    """, unsafe_allow_html=True)
-                
-                with col4:
-                    st.markdown(f"""
-                    <div class="stat-box">
-                        <h2 style="color: #6c757d;">📰 {results['count']}</h2>
-                        <p>Total Sources</p>
-                    </div>
-                    """, unsafe_allow_html=True)
+                col1, col2, col3, col4 = st.columns(4)
+                with col1: st.markdown(f"""<div class="stat-box"><h2 style="color:#28a745;">✅ {true_count}</h2><p>True/Accurate</p></div>""", unsafe_allow_html=True)
+                with col2: st.markdown(f"""<div class="stat-box"><h2 style="color:#dc3545;">❌ {false_count}</h2><p>False/Incorrect</p></div>""", unsafe_allow_html=True)
+                with col3: st.markdown(f"""<div class="stat-box"><h2 style="color:#ffc107;">⚠️ {mixed_count}</h2><p>Mixed/Misleading</p></div>""", unsafe_allow_html=True)
+                with col4: st.markdown(f"""<div class="stat-box"><h2 style="color:#6c757d;">📰 {results['count']}</h2><p>Total Sources</p></div>""", unsafe_allow_html=True)
                 
                 st.markdown("---")
-                
-                st.markdown(
-                    '<div class="section-title">🔍 Detailed Fact-Check Results</div>',
-                    unsafe_allow_html=True
-                )
+                st.markdown('<div class="section-title">🔍 Detailed Fact-Check Results</div>', unsafe_allow_html=True)
                 
                 for idx, result in enumerate(results["results"], 1):
                     verdict_class = determine_verdict_color(result["rating"])
                     emoji = get_verdict_emoji(result["rating"])
-                    
                     st.markdown(f"""
                     <div class="{verdict_class}">
                         <h3>{emoji} Fact-Check #{idx}</h3>
@@ -493,36 +399,12 @@ def main():
                     </div>
                     """, unsafe_allow_html=True)
                 
-                # Export options
+                # Export Report
                 st.markdown("---")
-                st.markdown(
-                    '<div class="section-title">📄 Report</div>',
-                    unsafe_allow_html=True
-                )
-                
-                # Create export data
-                export_text = f"""VERIFACT - CLAIM VERIFICATION REPORT
-Generated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
-
-CLAIM: {claim_text}
-
-SUMMARY:
-- Total Fact-Checks: {results['count']}
-- True/Accurate: {true_count}
-- False/Incorrect: {false_count}
-- Mixed/Misleading: {mixed_count}
-
-DETAILED RESULTS:
-"""
-                for idx, result in enumerate(results["results"], 1):
-                    export_text += f"""
-{idx}. {result["publisher"]}
-   Rating: {result["rating"]}
-   URL: {result["url"]}
-   Claimant: {result["claimant"]}
-   Date: {result["claim_date"]}
-   Title: {result["title"]}
-"""
+                st.markdown('<div class="section-title">📄 Report</div>', unsafe_allow_html=True)
+                export_text = f"VERIFACT - CLAIM VERIFICATION REPORT\nGenerated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n\nCLAIM: {claim_text}\n\nSUMMARY:\n- Total Fact-Checks: {results['count']}\n- True/Accurate: {true_count}\n- False/Incorrect: {false_count}\n- Mixed/Misleading: {mixed_count}\n\nDETAILED RESULTS:\n"
+                for idx, result in enumerate(results["results"],1):
+                    export_text += f"\n{idx}. {result['publisher']}\n   Rating: {result['rating']}\n   URL: {result['url']}\n   Claimant: {result['claimant']}\n   Date: {result['claim_date']}\n   Title: {result['title']}\n"
                 
                 st.download_button(
                     label="📄 Download Report",
@@ -535,13 +417,12 @@ DETAILED RESULTS:
     # Footer
     st.markdown("---")
     st.markdown("""
-    <div style="text-align: center; color: #666; padding: 2rem 0;">
-        <p style="font-size: 1.1rem;"><strong>VeriFact</strong> | Powered by Google Fact Check Tools API</p>
+    <div style="text-align:center;color:#666;padding:2rem 0;">
+        <p style="font-size:1.1rem;"><strong>VeriFact</strong> | Powered by Google Fact Check Tools API</p>
         <p>Built by <strong>Areeba Fatima</strong> with Python & Streamlit</p>
-        <p style="font-size: 0.85rem; color: #999;">⚠️ Always verify important claims from multiple sources. This tool is for informational purposes.</p>
+        <p style="font-size:0.85rem;color:#999;">⚠️ Always verify important claims from multiple sources. This tool is for informational purposes.</p>
     </div>
     """, unsafe_allow_html=True)
 
-
 if __name__ == "__main__":
     main()