Update app.py

app.py

@@ -3,11 +3,12 @@ import pandas as pd
 import requests
 import tempfile
 from docx import Document
+from datetime import datetime
 
-# DuckDuckGo search API
+# DuckDuckGo API search
 def search_api(query):
     if not query.strip():
-        return pd.DataFrame([{"Title": "Empty query. Please enter a topic.", "URL": ""}])
+        return pd.DataFrame([{"Rank": "", "Title": "Please enter a valid search query.", "URL": ""}])
     
     url = f"https://api.duckduckgo.com/?q={query}&format=json&no_redirect=1"
     try:
@@ -16,89 +17,90 @@ def search_api(query):
         data = response.json()
         related_topics = data.get("RelatedTopics", [])
         results = []
-        
+
         for topic in related_topics:
             if "Text" in topic and "FirstURL" in topic:
-                results.append({
-                    "Title": topic["Text"],
-                    "URL": topic["FirstURL"]
-                })
+                results.append({"Title": topic["Text"], "URL": topic["FirstURL"]})
             elif "Topics" in topic:
-                for subtopic in topic["Topics"]:
-                    if "Text" in subtopic and "FirstURL" in subtopic:
-                        results.append({
-                            "Title": subtopic["Text"],
-                            "URL": subtopic["FirstURL"]
-                        })
-        
+                for sub in topic["Topics"]:
+                    if "Text" in sub and "FirstURL" in sub:
+                        results.append({"Title": sub["Text"], "URL": sub["FirstURL"]})
+
         if not results:
             results.append({"Title": "No results found.", "URL": ""})
 
-        return pd.DataFrame(results)
-    
+        # Add ranking
+        for i, r in enumerate(results):
+            r["Rank"] = i + 1
+
+        return pd.DataFrame(results)[["Rank", "Title", "URL"]]
     except Exception as e:
-        return pd.DataFrame([{"Title": "Error fetching results.", "URL": str(e)}])
+        return pd.DataFrame([{"Rank": "", "Title": "Error fetching results.", "URL": str(e)}])
 
-# Export to CSV
-def export_csv(results):
+# Export CSV with timestamp
+def export_csv(results, query):
     if results is None or results.empty:
         return None
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    fname = f"{query.replace(' ', '_')}_results_{timestamp}.csv"
     with tempfile.NamedTemporaryFile(delete=False, suffix=".csv", mode="w", newline='', encoding="utf-8") as tmp:
         results.to_csv(tmp.name, index=False)
         return tmp.name
 
-# Export to DOCX
-def export_docx(results):
+# Export DOCX with timestamp
+def export_docx(results, query):
     if results is None or results.empty:
         return None
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    fname = f"{query.replace(' ', '_')}_results_{timestamp}.docx"
     doc = Document()
-    doc.add_heading("Search Results", 0)
-    for _, row in results.iterrows():
-        doc.add_paragraph(f"{row['Title']}\n{row['URL']}")
+    doc.add_heading(f"Search Results for: {query}", 0)
+    for i, row in results.iterrows():
+        doc.add_paragraph(f"{row['Rank']}. {row['Title']}\n{row['URL']}")
     with tempfile.NamedTemporaryFile(delete=False, suffix=".docx") as tmp:
         doc.save(tmp.name)
         return tmp.name
 
-# Gradio UI
+# UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🔍 DuckDuckGo Search & Export Tool")
+    gr.Markdown("## 🔎 Smart Search & Export App")
     gr.Markdown("""
-    Welcome to the DuckDuckGo Search & Export App!
-
-    **🔹 Instructions:**
-    1. Enter a **search query** in the textbox below.
-    2. Click **Search** to view relevant results from DuckDuckGo.
-    3. You can **download** the results as a **CSV file** or a **Word document**.
+    This app lets you search DuckDuckGo and export the results as a **CSV or Word DOCX**.
 
-    **🔹 Features:**
-    - Fast search via DuckDuckGo Instant Answer API
-    - Download results in CSV or DOCX format
-    - Mobile & desktop friendly
-
-    ⚠️ *Note: This tool fetches basic related links, not full web content.*
+    **Instructions:**
+    1. Type a search query (e.g., *climate change*).
+    2. Click **Search** to get top DuckDuckGo topics.
+    3. Review the list and click to open links.
+    4. Export results for offline use.
     """)
 
-    query = gr.Textbox(label="Search Query", placeholder="Enter a topic to search...")
-    search_btn = gr.Button("Search")
+    query = gr.Textbox(label="Search Query", placeholder="Enter topic...")
+    status = gr.Markdown("")
+    search_btn = gr.Button("🔍 Search Now")
 
-    result_state = gr.State(value=pd.DataFrame())
-    results_display = gr.Dataframe(headers=["Title", "URL"], label="Search Results", interactive=False)
+    results_display = gr.Dataframe(headers=["Rank", "Title", "URL"], label="Search Results", interactive=False)
 
     with gr.Row():
-        export_csv_btn = gr.Button("Export as CSV")
-        export_docx_btn = gr.Button("Export as Word DOCX")
+        export_csv_btn = gr.Button("⬇️ Export CSV")
+        export_docx_btn = gr.Button("⬇️ Export DOCX")
 
     csv_file = gr.File(label="Download CSV")
     docx_file = gr.File(label="Download DOCX")
 
-    def run_search(q):
+    def handle_search(q):
+        status_msg = f"Searching DuckDuckGo for **{q}**..."
         df = search_api(q)
-        return df, df
-
-    search_btn.click(fn=run_search, inputs=query, outputs=[results_display, result_state])
-    export_csv_btn.click(fn=export_csv, inputs=result_state, outputs=csv_file)
-    export_docx_btn.click(fn=export_docx, inputs=result_state, outputs=docx_file)
-
-demo.launch()
-
-
+        if "Error" in df.iloc[0]["Title"]:
+            status_msg = f"❌ Error: {df.iloc[0]['URL']}"
+        elif "Please enter" in df.iloc[0]["Title"]:
+            status_msg = "⚠️ Please enter a search term."
+        else:
+            status_msg = f"✅ Showing top {len(df)} results for: **{q}**"
+        return df, status_msg
+
+    # Actions
+    search_btn.click(fn=handle_search, inputs=query, outputs=[results_display, status])
+    export_csv_btn.click(fn=export_csv, inputs=[results_display, query], outputs=csv_file)
+    export_docx_btn.click(fn=export_docx, inputs=[results_display, query], outputs=docx_file)
+
+demo.launch(share=True)