Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import json
 import asyncio
+from datetime import datetime
 import gradio as gr
 import httpx
 
@@ -15,7 +16,12 @@ perplexity_client = PerplexityClient(
     api_key=os.environ["PERPLEXITY_AUTH_TOKEN"]
     )
 
-async def query_api(query: str) -> tuple:
+async def query_api(query: str, website: str, after_date: str) -> tuple:
+    if website:
+        query = query + f" site:{website}"
+    if after_date:
+        query = query + f" after:{after_date}"
+    print("query: ", query)
     try:
         # Fetch response from Perplexity
         response = await perplexity_client.generate_response(query)
@@ -62,7 +68,23 @@ with gr.Blocks(css=".gradio-container { background-color: #f5f5f5; padding: 20px
                 label="Enter your query",
                 placeholder="Type your search query here...",
                 lines=2,
-                max_lines=4,
+                max_lines=20,
+                value="",
+                elem_id="query-input"
+            )
+            website = gr.Textbox(
+                label="Single website search",
+                placeholder="website url",
+                lines=1,
+                max_lines=20,
+                value="",
+                elem_id="query-input"
+            )
+            after_date = gr.Textbox(
+                label="After date",
+                placeholder="date format: DD-MM-YYYY",
+                lines=1,
+                max_lines=2,
                 value="",
                 elem_id="query-input"
             )
@@ -91,7 +113,7 @@ with gr.Blocks(css=".gradio-container { background-color: #f5f5f5; padding: 20px
             )
     
     # Set up event listener
-    submit_button.click(query_api, inputs=query, outputs=[output_content, output_citations])
+    submit_button.click(query_api, inputs=[query, website, after_date], outputs=[output_content, output_citations])
     
     gr.Markdown("Powered by FastAPI and Gradio")