Update app.py

app.py

@@ -1,11 +1,12 @@
+import gradio as gr
 import requests
 import os
 
-# Hugging Face API URL for LegalBERT model (replace with your model URL)
+# Set the Hugging Face API URL for your model
 API_URL = "https://api-inference.huggingface.co/models/nlpaueb/legal-bert-base-uncased"
-API_KEY = os.getenv("HF_API_KEY")  # Fetch API key securely from environment variable
+API_KEY = os.getenv("HF_API_KEY")  # Securely fetch the API key from an environment variable
 
-# Function to send contract to Hugging Face API and get analysis results
+# Function to send contract text to Hugging Face API and get analysis results
 def analyze_contract(contract_text):
     headers = {
         "Authorization": f"Bearer {API_KEY}",
@@ -14,39 +15,46 @@ def analyze_contract(contract_text):
     
     payload = {"inputs": contract_text}
     
-    try:
-        # Send the contract text to Hugging Face for analysis
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=60)  # Increased timeout
-        
-        if response.status_code == 200:
+    # Send the contract text to Hugging Face for analysis
+    response = requests.post(API_URL, headers=headers, json=payload)
+
+    if response.status_code == 200:
+        try:
             result = response.json()
 
             # Print the raw response for debugging
             print("Raw Response from Hugging Face:", result)
 
             # Extracting the risk score, high-risk clauses, and the risk map URL
+            # (The actual keys here may vary depending on how your model is trained)
             overall_risk_score = result.get('overall_score', 'No score available')
             high_risk_clauses = result.get('high_risk_clauses', 'No high-risk clauses detected')
             risk_map_url = result.get('risk_map_url', 'No risk map available')
 
             return overall_risk_score, high_risk_clauses, risk_map_url
 
-        else:
-            print(f"Error: Unable to analyze the contract, Status Code: {response.status_code} - {response.text}")
-            return "Error: Unable to analyze the contract", "", ""
-
-    except requests.exceptions.ConnectionError as e:
-        print(f"Connection Error: {e}")
-        return "Error: Unable to connect to Hugging Face API", "", ""
-
-    except requests.exceptions.Timeout as e:
-        print(f"Timeout Error: {e}")
-        return "Error: Timeout while connecting to Hugging Face API", "", ""
-
-    except Exception as e:
-        print(f"General Error: {e}")
-        return f"Error: {str(e)}", "", ""
-
-# Test the function with contract text
-contract_text = "This is a sample contract. The termination clause should be reviewed carefully. The indemnity clause has potential risks."
-analyze_contract(contract_text)
+        except Exception as e:
+            return f"Error in processing response: {str(e)}", "", ""
+    else:
+        return f"Error: Unable to analyze the contract, Status Code: {response.status_code} - {response.text}", "", ""
+
+# Gradio UI setup
+def create_ui():
+    # Gradio interface
+    with gr.Blocks() as demo:
+        gr.Markdown("### Contract Risk Analyzer")
+        with gr.Row():
+            contract_input = gr.Textbox(label="Contract Text", placeholder="Paste contract text here", lines=10)
+        with gr.Row():
+            analyze_button = gr.Button("Analyze Contract")
+        with gr.Row():
+            risk_score_output = gr.Textbox(label="Overall Risk Score", interactive=False)
+            high_risk_clauses_output = gr.Textbox(label="High Risk Clauses", interactive=False)
+            risk_map_url_output = gr.Textbox(label="Risk Map URL", interactive=False)
+
+        analyze_button.click(fn=analyze_contract, inputs=contract_input, outputs=[risk_score_output, high_risk_clauses_output, risk_map_url_output])
+
+    demo.launch()
+
+if __name__ == "__main__":
+    create_ui()