Update app.py

app.py

@@ -1,37 +1,34 @@
+import gradio as gr
 import requests
-import json
+import os
 
-# Replace with your Hugging Face API Key and Model URL
-API_URL = "https://api-inference.huggingface.co/models/bert-base-uncased"  # Change this URL to your model
-API_KEY = "hf_YOUR_API_KEY"  # Replace this with your actual Hugging Face API key
+# Hugging Face API URL for LegalBERT model (replace with your model URL)
+API_URL = "https://api-inference.huggingface.co/models/joeddav/legal-bert"  # LegalBERT example URL
+API_KEY = os.getenv("HF_API_KEY")  # Fetch API key securely from environment variable
 
-# Function to send contract text to Hugging Face API and get analysis results
+# Function to send contract to Hugging Face API and get analysis results
 def analyze_contract(contract_text):
     headers = {
         "Authorization": f"Bearer {API_KEY}",
         "Content-Type": "application/json"
     }
-
-    # Prepare the payload with the contract text
-    payload = {
-        "inputs": contract_text
-    }
-
-    # Send the request to Hugging Face
+    
+    payload = {"inputs": contract_text}
+    
+    # Send the contract text to Hugging Face for analysis
     response = requests.post(API_URL, headers=headers, json=payload)
 
     if response.status_code == 200:
         try:
-            # Parse the response JSON
             result = response.json()
 
             # Print the raw response for debugging
             print("Raw Response from Hugging Face:", result)
 
-            # Example: Extracting fields from the response (based on your expected output structure)
-            # If the model returns 'overall_score', 'high_risk_clauses', and 'risk_map_url', you can parse it here
+            # 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', [])
+            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
@@ -39,20 +36,25 @@ def analyze_contract(contract_text):
         except Exception as e:
             return f"Error in processing response: {str(e)}", "", ""
     else:
-        # Handle the error if API request fails
         return f"Error: Unable to analyze the contract, Status Code: {response.status_code} - {response.text}", "", ""
 
-# Example contract text for testing
-contract_text = """
-This Agreement is made between Party A (Seller) and Party B (Buyer). 
-1. Termination Clause: The agreement can be terminated with a 30-day written notice.
-2. Indemnity Clause: Party B agrees to indemnify Party A for any losses.
-"""
-
-# Run the analysis
-risk_score, high_risk_clauses, risk_map_url = analyze_contract(contract_text)
-
-# Print the results
-print("Overall Risk Score:", risk_score)
-print("High Risk Clauses:", high_risk_clauses)
-print("Risk Map URL:", risk_map_url)
+# 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()