Update app.py

app.py

@@ -1,11 +1,10 @@
-import os
+import gradio as gr
 import requests
+import os
 
-# Hugging Face API URL for the LegalBERT model
 API_URL = "https://api-inference.huggingface.co/models/nlpaueb/legal-bert-base-uncased"
-API_KEY = os.getenv("Contract_Risk")  # Fetch the API key securely from environment variables
+API_KEY = os.getenv("Contract_Risk")
 
-# Function to send contract to Hugging Face API and get analysis results
 def analyze_contract(contract_text):
     headers = {
         "Authorization": f"Bearer {API_KEY}",
@@ -17,15 +16,10 @@ def analyze_contract(contract_text):
     try:
         # Send the contract text to Hugging Face for analysis
         response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
-        response.raise_for_status()  # Raise an exception for invalid responses (4xx/5xx)
+        response.raise_for_status()
 
-        # Parse the response from Hugging Face
         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 (if available)
         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')
@@ -41,3 +35,22 @@ def analyze_contract(contract_text):
     except Exception as e:
         print(f"Unexpected error: {e}")
         return f"Unexpected error: {e}", "", ""
+
+def create_ui():
+    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()