Create main.py

main.py

@@ -0,0 +1,35 @@
+from fastapi import FastAPI, HTTPException
+from typing import Dict
+import logging
+import httpx
+
+app = FastAPI()
+
+logging.basicConfig(level=logging.DEBUG)
+
+# 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
+async def summarize_file_with_gradio(file_content: str) -> Dict[str, str]:
+    try:
+        async with httpx.AsyncClient() as client:
+            response = await client.post(f"{gradio_api_url}/predict", json={"file_content": file_content})
+            if response.status_code == 200:
+                result = response.json()
+                return {"summary": result[0]}
+            else:
+                return {"error": "Gradio API request failed"}
+    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
+
+if __name__ == '__main__':
+    import uvicorn
+    uvicorn.run(app)