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