Upload factchecking_front.py

factchecking_front.py

@@ -1,14 +1,55 @@
 import gradio as gr
-from gradio_client import Client
-import os 
+import os
+import requests
+import time
 
-client = Client("Iker/FactChecking-Backend")
+API_URL = "https://Iker-FactChecking-Backend.hf.space:7860"  # Replace with your server's URL if different
+API_KEY = os.getenv("API_KEY")
 
+headers = {"X-API-Key": API_KEY, "Content-Type": "application/json"}
 
-def fact_checking(
-    article_topic: str,
-):
-    result = client.predict(article_topic=article_topic, api_name="/partial")
+
+# Function to submit a fact-checking request
+def submit_fact_check(article_topic, config):
+    endpoint = f"{API_URL}/fact-check"
+    payload = {"article_topic": article_topic, "config": config}
+
+    response = requests.post(endpoint, json=payload, headers=headers)
+    response.raise_for_status()  # Raise an exception for HTTP errors
+    return response.json()["job_id"]
+
+
+# Function to get the result of a fact-checking job
+def get_fact_check_result(job_id):
+    endpoint = f"{API_URL}/result/{job_id}"
+
+    response = requests.get(endpoint, headers=headers)
+    response.raise_for_status()  # Raise an exception for HTTP errors
+    return response.json()
+
+
+def fact_checking(article_topic: str, config="pro"):
+    try:
+        job_id = submit_fact_check(article_topic, config.lower())
+        print(f"Fact-checking job submitted. Job ID: {job_id}")
+
+        # Poll for results
+        while True:
+            result = get_fact_check_result(job_id)
+            if result["status"] == "completed":
+                print("Fact-checking completed:")
+                print(result["result"])
+                break
+            elif result["status"] == "failed":
+                print("Fact-checking failed:")
+                print(result["error"])
+                break
+            else:
+                print("Fact-checking in progress. Waiting...")
+                time.sleep(2)  # Wait for 2 seconds before checking again
+
+    except requests.exceptions.RequestException as e:
+        print(f"An error occurred: {e}")
 
     return result
 
@@ -53,6 +94,16 @@ if __name__ == "__main__":
             placeholder="Los coches electricos contaminan más que los coches de gasolina",
         )
 
+        gr_mode = gr.Radio(
+            label="Fact-Checking Mode",
+            info="Choose the fact-checking mode. The \"Turbo\" mode is faster and cheaper. The \"Pro\" mode is more accurate but more expensive.",
+            choices=["Pro", "Turbo"],
+            type="value",
+            default="Pro",
+            show_label=False,
+            interactive=True,
+        )
+
         gr_play = gr.Button("Fact-Checking")
 
         gr_output = gr.Markdown(
@@ -93,7 +144,7 @@ if __name__ == "__main__":
 
         gr_play.click(
             fact_checking,
-            inputs=[gr_text],
+            inputs=[gr_text,gr_mode],
             outputs=[gr_output, gr_ft, gr_qa, gr_citations, gr_image],
         )