Update app.py

app.py

@@ -1,64 +1,82 @@
 import gradio as gr
 import requests
+import json
 
 API_URL = "https://sendthat.cc"
 
+def list_indexes():
+ url = f"{API_URL}/list_indexes"
+ response = requests.get(url)
+ return response.json()
+
 def search_document(index_name, query, k):
  url = f"{API_URL}/search/{index_name}"
  payload = {"text": query, "k": k}
  headers = {"Content-Type": "application/json"}
  response = requests.post(url, json=payload, headers=headers)
- results = response.json()
- 
- formatted_results = []
- for result in results.get('results', []):
- metadata = result.get('metadata', {})
- formatted_result = f"Source: {metadata.get('source', 'Unknown')}\n"
- formatted_result += f"Page: {metadata.get('page', 'Unknown')}\n"
- formatted_result += f"Content: {metadata.get('content', 'No content available')}\n"
- formatted_result += f"Distance: {result.get('distance', 'Unknown')}\n"
- formatted_results.append(formatted_result)
- 
- return "\n\n".join(formatted_results)
+ return response.json(), "", k
 
 def qa_document(index_name, question, k):
  url = f"{API_URL}/qa/{index_name}"
  payload = {"text": question, "k": k}
  headers = {"Content-Type": "application/json"}
  response = requests.post(url, json=payload, headers=headers)
- result = response.json()
- 
- answer = result.get('answer', 'No answer available')
- sources = result.get('sources', [])
- 
- formatted_sources = []
- for source in sources:
- formatted_source = f"Source: {source.get('source', 'Unknown')}\n"
- formatted_source += f"Relevance Score: {source.get('relevance_score', 'Unknown')}"
- formatted_sources.append(formatted_source)
- 
- formatted_result = f"Answer: {answer}\n\nSources:\n" + "\n\n".join(formatted_sources)
- return formatted_result
+ return response.json(), "", k
+
+def refresh_indexes():
+ indexes = list_indexes()
+ return gr.Dropdown(choices=indexes, label="Select Index")
 
-with gr.Blocks() as demo:
+# Custom CSS for modern appearance
+custom_css = """
+.gradio-container {
+ background-color: #f5f5f5;
+}
+.input-box, .output-box, .gr-box {
+ border-radius: 8px !important;
+ border-color: #d1d1d1 !important;
+}
+.input-box, .gr-dropdown {
+ background-color: #e6f2ff !important;
+}
+.gr-button {
+ background-color: #4a90e2 !important;
+ color: white !important;
+}
+.gr-button:hover {
+ background-color: #3a7bc8 !important;
+}
+.gr-form {
+ border-color: #d1d1d1 !important;
+ background-color: white !important;
+}
+"""
+
+with gr.Blocks(css=custom_css) as demo:
  gr.Markdown("# Document Search and Question Answering System")
  
- index_name = gr.Textbox(label="Index Name", value="default")
- 
+ with gr.Row():
+ index_dropdown = gr.Dropdown(label="Select Index", choices=list_indexes())
+ refresh_button = gr.Button("Refresh Indexes")
+
  with gr.Tab("Search"):
  search_input = gr.Textbox(label="Search Query")
  search_k = gr.Slider(1, 10, 5, step=1, label="Number of Results")
  search_button = gr.Button("Search")
- search_output = gr.Textbox(label="Search Results", lines=10)
+ search_output = gr.JSON(label="Search Results")
  
- search_button.click(search_document, inputs=[index_name, search_input, search_k], outputs=search_output)
- 
  with gr.Tab("Question Answering"):
  qa_input = gr.Textbox(label="Question")
  qa_k = gr.Slider(1, 10, 5, step=1, label="Number of Contexts to Consider")
  qa_button = gr.Button("Ask Question")
- qa_output = gr.Textbox(label="Answer and Sources", lines=10)
- 
- qa_button.click(qa_document, inputs=[index_name, qa_input, qa_k], outputs=qa_output)
+ qa_output = gr.JSON(label="Answer")
+ 
+ refresh_button.click(refresh_indexes, outputs=index_dropdown)
+ search_button.click(search_document, 
+ inputs=[index_dropdown, search_input, search_k], 
+ outputs=[search_output, search_input, search_k])
+ qa_button.click(qa_document, 
+ inputs=[index_dropdown, qa_input, qa_k], 
+ outputs=[qa_output, qa_input, qa_k])
 
 demo.launch()