Update main.py

main.py

@@ -1,25 +1,21 @@
-from fastapi import FastAPI, HTTPException
-from typing import Dict
-import httpx
+from flask import Flask, request, jsonify
+from gradio_client import Client
+from flask_cors import CORS, cross_origin
 
-app = FastAPI()
+app = Flask(__name__)
+cors = CORS(app)
 
 # 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
+@app.route('/summarize', methods=['GET' , 'POST'])
+def summarize_text():
+    if request.method == 'POST':
+        data = request.get_json()
+        file_content = data.get('file_content')
+        # app.logger.info(file_content)
+        result = summarize_file_with_gradio(file_content)
+        return jsonify(result)