app.py.old

import gradio as gr
import requests
import json
import logging

logging.basicConfig(level=logging.DEBUG)

API_URL = "https://sendthat.cc"

def list_indexes():
 try:
 url = f"{API_URL}/list_indexes"
 response = requests.get(url)
 response.raise_for_status()
 indexes = response.json()
 
 logging.debug(f"API response for indexes: {indexes}")
 
 if isinstance(indexes, list):
 if "indexes" in indexes:
 return ["history", "medical"] # Hardcoded index names
 elif indexes:
 return indexes
 else:
 return ["No indexes available"]
 elif isinstance(indexes, dict) and "indexes" in indexes:
 return indexes["indexes"] if indexes["indexes"] else ["No indexes available"]
 else:
 logging.warning(f"Unexpected response format: {indexes}")
 return ["history", "medical"] # Fallback to hardcoded index names
 except requests.exceptions.RequestException as e:
 logging.error(f"Error fetching indexes: {e}")
 return ["Error fetching indexes"]
 except json.JSONDecodeError as e:
 logging.error(f"Error decoding JSON: {e}")
 return ["Error decoding response"]

def search_document(index_name, query, k):
 try:
 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)
 response.raise_for_status()
 return response.json(), "", k
 except requests.exceptions.RequestException as e:
 logging.error(f"Error in search: {e}")
 return {"error": str(e)}, query, k

def qa_document(index_name, question, k):
 try:
 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)
 response.raise_for_status()
 return response.json(), "", k
 except requests.exceptions.RequestException as e:
 logging.error(f"Error in QA: {e}")
 return {"error": str(e)}, question, k

def refresh_indexes():
 indexes = list_indexes()
 return gr.Dropdown(choices=indexes, label="Select Index")

# 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")
 
 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.JSON(label="Search Results")
 
 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.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()