Upload factchecking_front.py

factchecking_front.py

@@ -1,55 +1,15 @@
-import gradio as gr
 import os
-import requests
-import time
-
-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"}
-
-
-# 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}"
+import gradio as gr
+from gradio_client import Client
 
-    response = requests.get(endpoint, headers=headers)
-    response.raise_for_status()  # Raise an exception for HTTP errors
-    return response.json()
+client = Client("Iker/FactChecking-Backend")
 
 
 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}")
+    result = client.predict(
+        article_topic=article_topic, config=config, api_name="/partial"
+    )
 
     return result
 
@@ -96,7 +56,7 @@ if __name__ == "__main__":
 
         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.",
+            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",
             value="Pro",
@@ -144,7 +104,7 @@ if __name__ == "__main__":
 
         gr_play.click(
             fact_checking,
-            inputs=[gr_text,gr_mode],
+            inputs=[gr_text, gr_mode],
             outputs=[gr_output, gr_ft, gr_qa, gr_citations, gr_image],
         )