import gradio as gr
import os
import requests
def query_huggingface(payload):
    try:
        response = requests.post(API_URL, headers=HEADERS, json=payload)
        response.raise_for_status()  # Raise exception for non-2xx status codes
        return response.json()
    except requests.exceptions.RequestException as e:
        print(f"Error querying Hugging Face API: {e}")
        return {"summary_text": f"Error querying Hugging Face API: {e}"}

def respond(user_message):
    # Construct input text for summarization (only user message)
    input_text = f"User: {user_message}"

    # Query Hugging Face API for summarization
    payload = {"inputs": input_text}
    response = query_huggingface(payload)

    # Extract summary text from the API response
    summary_text = summary_text = response[0]["summary_text"] if isinstance(response, list) else response.get("summary_text", "")


    return summary_text

# Launch Gradio chat interface with the predict function
gr.ChatInterface(respond).launch()