Update app.py

app.py

@@ -1,12 +1,6 @@
-import gradio as gr
-import requests
+from flask import Flask, render_template, request
 
-API_URL = "https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct"
-headers = {"Authorization": "Bearer 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"}
-
-def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+app = Flask(__name__)
 
 API_URL2 = "https://api-inference.huggingface.co/models/valhalla/longformer-base-4096-finetuned-squadv1"
 headers2 = {"Authorization": "Bearer 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"}
@@ -15,45 +9,20 @@ def query2(payload):
     response = requests.post(API_URL2, headers=headers2, json=payload)
     return response.json()
 
-def ask_question(question, context):
-    output2 = query2({
-        "inputs": {
-            "question": question,
-            "context": context
-        },
-    })
-    return output2
-
-iface_ask = gr.Interface(
-    fn=ask_question,
-    inputs=[
-        gr.Textbox(type="text", placeholder="Enter your question"),
-        gr.Textbox(type="text", placeholder="Enter context"),
-        gr.Button("Ask", onclick="update_output()")  # Use custom JavaScript function
-    ],
-    outputs=gr.Textbox(type="text", placeholder="Answer"),  # Single Textbox for the answer
-    live=False  # Set live to False
-)
-
-# Define the JavaScript function
-javascript_code = """
-function update_output() {
-    var question = gr("#text-input-0").val();
-    var context = gr("#text-input-1").val();
-    
-    // Call the backend function with the updated values
-    gr('iframe').contentWindow.postMessage({
-        'task': 'updateOutput',
-        'data': {
-            'question': question,
-            'context': context
-        }
-    }, '*');
-}
-"""
-
-# Attach the JavaScript code to the interface
-iface_ask.inferface.html("head", f"<script>{javascript_code}</script>")
-
-iface_ask.launch()
-
+@app.route("/", methods=["GET", "POST"])
+def home():
+    answer = ""
+    if request.method == "POST":
+        question = request.form["question"]
+        context = request.form["context"]
+        output2 = query2({
+            "inputs": {
+                "question": question,
+                "context": context
+            },
+        })
+        answer = output2["answer"]
+    return render_template("index.html", answer=answer)
+
+if __name__ == "__main__":
+    app.run(debug=True)