import gradio as gr
import requests
import json

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()
        
        # Check if the response is just ["indexes"] and replace with actual index names
        if indexes == ["indexes"]:
            return ["history", "medical"]  # Hardcoded index names
        elif isinstance(indexes, list) and indexes:
            return indexes
        else:
            return ["No indexes available"]
    except requests.exceptions.RequestException as e:
        print(f"Error fetching indexes: {e}")
        return ["Error fetching indexes"]

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)
    return response.json(), "", k  # Return empty string to clear input

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)
    return response.json(), "", k  # Return empty string to clear input

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()