Update main.py

main.py

@@ -1,21 +1,20 @@
-from flask import Flask, request, jsonify
+from fastapi import FastAPI
 from gradio_client import Client
-from flask_cors import CORS, cross_origin
 
-app = Flask(__name__)
-cors = CORS(app)
+app = FastAPI()
 
-# Define the Gradio API endpoint
-gradio_api_url = "https://astro21-test-2.hf.space/--replicas/x5m8s/"
+# Define a route for the prediction using FastAPI
+@app.post("/predict")
+async def predict(text: str):
+    # Replace this URL with the actual API endpoint URL
+    api_endpoint_url = "https://astro21-test-2.hf.space/--replicas/x5m8s/"
 
-# Function to summarize a file using the Gradio API
+    # Use the Gradio client to make the prediction request
+    client = Client(api_endpoint_url)
+    result = client.predict(
+        text,
+        api_name="/predict"
+    )
 
-# Define a route to handle summarization requests
-@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)
+    # Return the result as a response
+    return {"result": result}