from fastapi import FastAPI, HTTPException
from typing import Dict
import httpx

app = FastAPI()

# Define the Gradio API endpoint
gradio_api_url = "https://astro21-test-2.hf.space/--replicas/x5m8s/"

# Function to summarize a file using the Gradio API
# Function to summarize a file using the Gradio API
async def summarize_file_with_gradio(file_content: str) -> Dict[str, str]:
    try:
        client = GradioClient(gradio_api_url)
        result = await client.predict(file_content, api_name="/predict")
        return {"summary": result[0]}
    except Exception as e:
        return {"error": str(e)}

# Define a route to handle summarization requests
@app.post('/summarize', response_model=Dict[str, str])
async def summarize_text(data: Dict[str, str]):
    file_content = data.get('file_content')
    result = await summarize_file_with_gradio(file_content)
    return result