Update app.py

app.py

@@ -6,18 +6,18 @@ from langchain.docstore.document import Document
 from datasets import load_dataset, concatenate_datasets
 
 st.set_page_config(
-    page_title="African History Search Engine",
-    page_icon="🌍",
+    page_title="Science Search Engine",
+    page_icon="📚",
     layout="wide"
 )
 
 class RetrieverTool(Tool):
     name = "retriever"
-    description = "Uses BM25 search to retrieve relevant African historical documentation"
+    description = "Uses BM25 search to retrieve relevant scientific documentation"
     inputs = {
         "query": {
             "type": "string",
-            "description": "The historical query in affirmative form rather than a question"
+            "description": "The scientific query in affirmative form rather than a question"
         }
     }
     output_type = "string"
@@ -102,20 +102,19 @@ def format_search_results(results: str):
 def get_agent():
     """Single function to handle data loading, processing, and agent creation"""
     # Load dataset
-    dataset = load_dataset("Svngoku/African-History-Extra-11-30-24")
+    dataset = load_dataset("camel-ai/biology")
     train_docs = dataset["train"]
-    test_docs = dataset["test"]
-    source_docs = concatenate_datasets([train_docs, test_docs])
+    source_docs = concatenate_datasets([train_docs])
     
     # Create documents
     documents = [
         Document(
-            page_content=item['content'],
+            page_content=item['message_2'],
             metadata={
-                "source": item['url'],
-                "title": item['title'],
-                "description": item['description'],
-                "published_time": item['publishedTime']
+              #  "source": item['url'],
+                "title": item['message_1'],
+                "description": item['sub_topic'],
+                # "published_time": item['publishedTime']
             }
         )
         for item in source_docs
@@ -138,41 +137,41 @@ def get_agent():
     )
 
 # Streamlit UI
-st.title("🌍 African History Search Engine")
+st.title("📚 Scientific Search Engine")
 st.markdown("""
-This search engine uses advanced AI to help you explore African history. 
-It provides detailed, sourced information from a curated database of historical documents.
+This search engine uses advanced AI to help you explore science. 
+It provides detailed, sourced information from a curated database of scientific knowledge.
 """)
 
 # Initialize agent
 if 'agent' not in st.session_state:
-    with st.spinner("Loading historical database..."):
+    with st.spinner("Loading database..."):
         st.session_state.agent = get_agent()
 
 # Search interface
 search_query = st.text_input(
     "🔍 Search African History",
-    placeholder="E.g., Tell me about the Kingdom of Kush",
-    help="Enter any question about African history"
+    placeholder="E.g., Tell me about cancer in dogs",
+    help="Enter any question about science"
 )
 
 # Advanced search options
 with st.expander("Advanced Search Options"):
     search_type = st.radio(
         "Search Type",
-        ["General Query", "Specific Time Period", "Geographic Region"],
+        ["General Query", "Scientific branches"],
         help="Select the type of search you want to perform"
     )
     
-    if search_type == "Specific Time Period":
-        search_query = f"Focus on the time period: {search_query}"
+    if search_type == "Scientific branches":
+        search_query = f"Focus on the specific scientific branch of: {search_query}"
     elif search_type == "Geographic Region":
         search_query = f"Focus on the region of: {search_query}"
 
 # Search button
 if st.button("Search", type="primary"):
     if search_query:
-        with st.spinner("Searching historical records..."):
+        with st.spinner("Searching records..."):
             try:
                 results = st.session_state.agent.run(search_query)
                 
@@ -199,7 +198,7 @@ with st.sidebar:
     st.markdown("### About This Search Engine")
     st.markdown("""
     This search engine specializes in African history, providing:
-    - 📚 Detailed historical information
+    - 📚 Detailed information
     - 🔍 Source verification
     - 🌍 Geographic context
     - ⏳ Historical timeline context
@@ -211,4 +210,4 @@ with st.sidebar:
 
 # Footer
 st.markdown("---")
-st.caption("Powered by SmolAgents, RAG, and African History Dataset")
+st.caption("Powered by SmolAgents, RAG, and Camel AI Dataset")