Upload app.py

app.py

@@ -0,0 +1,226 @@
+# -*- coding: utf-8 -*-
+"""app.ipynb
+
+Automatically generated by Colab.
+
+Original file is located at
+    https://colab.research.google.com/drive/1Yo-pZqBc-DxiTzP9TyKKFInBqeeQrlTb
+"""
+
+import os
+import json
+import asyncio
+from datetime import datetime
+from typing import Dict, List, Any
+
+import streamlit as st
+
+from helper import ChatBot, current_year, save_to_audio, invoke_duckduckgo_news_search
+
+# ============================ FRONT-END SETUP ============================
+
+st.set_page_config(layout="wide")  # Set Streamlit layout to wide mode
+st.title("SearchBot 🤖")  # App title
+
+# ============================ SIDEBAR SETTINGS ============================
+
+with st.sidebar:
+    with st.expander("📖 Instruction Manual"):
+        st.markdown(
+            """
+            ## 🧠 SearchBot 🤖 - Your AI-Powered Research Assistant
+            Welcome to **SearchBot**, an advanced AI assistant that helps you find the latest news, trends, and information
+            across various sources.
+
+            ### 🔹 How to Use:
+            1. **📌 Choose Search Source**
+               - Select the type of search (News, Research Papers, Web Articles).
+            2. **📊 Choose Number of Results**
+               - Decide how many results you want (1 to 10).
+            3. **🌍 Set Location**
+               - Customize search results based on location.
+               *(e.g., "us-en" for USA, "in-en" for India)*
+            4. **⏳ Filter by Time**
+               - Search for the most recent news or past articles:
+                 - **Past Day** 🕐 (Breaking News)
+                 - **Past Week** 🗓 (Trending Topics)
+                 - **Past Month** 📅 (Major Stories)
+                 - **Past Year** ���� (Deep Research)
+            5. **💬 Review Search Results & Chat History**
+               - View results in an interactive table.
+               - Chatbot provides summarized responses with references.
+
+            ---
+
+            ### 🔹 Live Examples You Can Try:
+            **📰 Find Latest News**
+            - *"What are the latest AI breakthroughs?"*
+            - *"Recent developments in space exploration."*
+
+            **📖 Research Papers & Analysis**
+            - *"Most cited papers on quantum computing."*
+            - *"Deep learning advancements in 2024."*
+
+            **🌍 Location-Based Information**
+            - *"Tech news in Silicon Valley."*
+            - *"Political updates in the UK."*
+
+            **⚡ AI-Powered Chatbot Insights**
+            - *"Summarize recent news on cryptocurrency."*
+            - *"Give me top AI news from last week with analysis."*
+
+            """
+        )
+
+    # User inputs for search customization
+    num: int = st.number_input("📊 Number of results", value=7, step=1, min_value=1, max_value=10)
+    location: str = st.text_input("🌍 Location (e.g., us-en, in-en)", value="us-en")
+    time_filter: str = st.selectbox(
+        "⏳ Time filter",
+        ["Past Day", "Past Week", "Past Month", "Past Year"],
+        index=1
+    )
+
+    # Convert time filter to DuckDuckGo-compatible format
+    time_mapping: Dict[str, str] = {"Past Day": "d", "Past Week": "w", "Past Month": "m", "Past Year": "y"}
+    time_filter = time_mapping[time_filter]
+
+    only_use_chatbot: bool = st.checkbox("💬 Only use chatbot (Disable Search)")
+
+    # Clear chat history button
+    if st.button("🧹 Clear Session"):
+        st.session_state.messages = []
+        st.rerun()
+
+    # Footer with dynamic year
+    st.markdown(f"<h6>📅 Copyright © 2010-{current_year()} Present</h6>", unsafe_allow_html=True)
+
+# ============================ CHAT HISTORY SETUP ============================
+
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages: List[Dict[str, str]] = []
+
+# Ensure messages are always a list of dictionaries
+if not isinstance(st.session_state.messages, list) or not all(isinstance(msg, dict) for msg in st.session_state.messages):
+    st.session_state.messages = []
+
+# Display past chat history in Streamlit chat UI
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+# ============================ CHAT INPUT & PROCESSING ============================
+
+# Process user input in the chatbox
+if prompt := st.chat_input("Ask anything!"):
+    st.chat_message("user").markdown(prompt)
+    st.session_state.messages.append({"role": "user", "content": prompt})
+
+    # **Initialize ref_table_string to hold search results**
+    ref_table_string: str = "**No references found.**"
+
+    try:
+        with st.spinner("Searching..."):  # Show loading spinner
+            if only_use_chatbot:
+                response: str = "<empty>"
+            else:
+                # **Call async search function using `asyncio.run()`**
+                search_results: Dict[str, Any] = asyncio.run(
+                    invoke_duckduckgo_news_search(query=prompt, location=location, num=num, time_filter=time_filter)
+                )
+
+                if search_results["status"] == "success":
+                    md_data: List[Dict[str, Any]] = search_results["results"]
+                    response = f"Here are your search results:\n{md_data}"
+
+                    def clean_title(title: str) -> str:
+                        """
+                        Cleans the title by replacing '|' with '-' to ensure proper formatting.
+
+                        Args:
+                            title (str): The original title.
+
+                        Returns:
+                            str: The cleaned title with '|' replaced by '-'.
+                        """
+                        return title.replace("|", " - ").strip()  # Replace '|' with ' - ' and remove leading/trailing spaces
+
+                    def generate_star_rating(rating: str) -> str:
+                        """
+                        Converts a numeric rating into a star representation (supports half-stars).
+
+                        Args:
+                            rating (str): The rating value as a string.
+
+                        Returns:
+                            str: A string representation of the rating using stars (⭐) and half-stars (⭐½).
+                        """
+                        try:
+                            rating_float: float = float(rating)  # Convert rating to float
+                            full_stars: int = int(rating_float)  # Extract full stars
+                            half_star: str = "⭐½" if (rating_float - full_stars) >= 0.5 else ""  # Add half-star if needed
+                            return "⭐" * full_stars + half_star  # Construct final star rating
+                        except ValueError:
+                            return "N/A"  # Fallback for non-numeric ratings
+
+                    # Start building reference table with proper Markdown formatting
+                    ref_table_string = "| Num | Title | Rating | Context |\n|---|------|--------|---------|\n"
+
+                    for res in md_data:
+                        # **Fix: Clean the title by replacing '|' with '-'**
+                        title_cleaned = clean_title(res['title'])
+
+                        # **Ensure the rating is always numeric before converting to stars**
+                        raw_rating = str(res.get('rating', 'N/A')).strip()  # Get rating and strip whitespace
+
+                        # Fix: Only convert rating if it’s a valid number
+                        if raw_rating.replace('.', '', 1).isdigit():  # Check if it’s a valid float
+                            stars = generate_star_rating(raw_rating)
+                        else:
+                            stars = "N/A"  # If it's text (like "MIT News"), default to "N/A"
+
+                        # **Ensure proper clickable links in the Title column**
+                        if res.get('link', '').startswith("http"):  # Ensure link exists and is valid
+                            title = f"[{title_cleaned}]({res['link']})"
+                        else:
+                            title = title_cleaned  # Fallback to text-only title
+
+                        # **Properly format Context column (limit to 100 chars)**
+                        context_summary = res.get('summary', '').strip()  # Ensure it's a string and strip spaces
+                        summary = context_summary[:100] + "..." if len(context_summary) > 100 else context_summary
+
+                        # **Final row construction**
+                        ref_table_string += f"| {res['num']} | {title} | {stars} | {summary} |\n"
+
+            # **Generate chatbot response based on search results or chat history**
+            bot = ChatBot()
+            bot.history = st.session_state.messages.copy()
+            response = bot.generate_response(
+                f"""
+                User prompt: {prompt}
+                Search results: {response}
+                Context: {[res['summary'] for res in search_results.get("results", [])]}
+                If search results exist, use them for the answer.
+                Otherwise, generate a response based on chat history.
+                """
+            )
+
+    except Exception as e:
+        st.warning(f"Error fetching data: {e}")
+        response = "We encountered an issue. Please try again later."
+
+    # **Convert response to audio**
+    save_to_audio(response)
+
+    # **Display assistant response in chat UI**
+    with st.chat_message("assistant"):
+        st.markdown(response, unsafe_allow_html=True)
+        st.audio("output.mp3", format="audio/mpeg", loop=True)
+        with st.expander("References:", expanded=True):
+            st.markdown(ref_table_string, unsafe_allow_html=True)
+
+    # **Update chat history with final response**
+    final_response: str = f"{response}\n\n{ref_table_string}"
+    st.session_state.messages.append({"role": "assistant", "content": final_response})
+