Update app.py

app.py

@@ -10,17 +10,16 @@ import fitz  # PyMuPDF
 import requests
 import uuid
 import io
+import difflib
 
 # Configuration
 MODEL_NAME = "google/gemma-2b-it"
 CURRENT_USER = "AkarshanGupta"
-CURRENT_TIME = "2025-03-23 04:53:36"
+CURRENT_TIME = "2025-03-23 05:05:24"
 
 # API Keys
 HF_TOKEN = os.getenv('HF_TOKEN')
 AZURE_TRANSLATION_KEY = os.getenv('AZURE_TRANSLATION_KEY')
-LLAMA_API_KEY = os.getenv('LLAMA_API_KEY')
-LLAMA_API_ENDPOINT = "https://api.llama.ai/v1/generate"
 
 class TextExtractor:
     @staticmethod
@@ -149,7 +148,6 @@ class LegalEaseAssistant:
         
         login(token=HF_TOKEN)
         
-        # Initialize text_extractor first
         self.text_extractor = TextExtractor()
         
         self.tokenizer = AutoTokenizer.from_pretrained(
@@ -175,7 +173,8 @@ class LegalEaseAssistant:
             "simplify": f"Simplify the following legal text in clear, plain language. Provide the response as separate points:\n\n{text}\n\nSimplified explanation:",
             "summary": f"Provide a concise summary of the following legal document as separate key points:\n\n{text}\n\nSummary:",
             "key_terms": f"Identify and explain the key legal terms and obligations in this text as separate points:\n\n{text}\n\nKey Terms:",
-            "risk": f"Perform a risk analysis on the following legal document and list each risk as a separate point:\n\n{text}\n\nRisk Assessment:"
+            "risk": f"Perform a risk analysis on the following legal document and list each risk as a separate point:\n\n{text}\n\nRisk Assessment:",
+            "compare": f"Compare the following contract sections and identify key differences:\n\n{text}\n\nDifferences:"
         }
         
         prompt = task_prompts.get(task_type, f"Analyze the following text and provide points:\n\n{text}\n\nAnalysis:")
@@ -195,23 +194,54 @@ class LegalEaseAssistant:
         raw_response = response_parts[-1].strip() if len(response_parts) > 1 else response.strip()
         
         return self.format_response(raw_response)
-    
-    def generate_chatbot_response(self, user_input):
-        if not LLAMA_API_KEY:
-            return "LLaMA API key not found. Please set the LLAMA_API_KEY environment variable."
+
+    def compare_contracts(self, contract1, contract2):
+        """
+        Compare two contracts and highlight the differences
+        """
+        text1 = self.text_extractor.extract_text_from_input(contract1)
+        text2 = self.text_extractor.extract_text_from_input(contract2)
         
-        response = requests.post(
-            LLAMA_API_ENDPOINT,
-            headers={"Authorization": f"Bearer {LLAMA_API_KEY}"},
-            json={"prompt": user_input, "max_tokens": 150}
-        )
+        # Split texts into lines
+        lines1 = text1.splitlines()
+        lines2 = text2.splitlines()
+        
+        # Generate the diff
+        differ = difflib.Differ()
+        diff = list(differ.compare(lines1, lines2))
+        
+        # Format the differences
+        differences = {
+            'added': [],
+            'removed': [],
+            'changed': []
+        }
+        
+        for line in diff:
+            if line.startswith('+ '):
+                differences['added'].append(line[2:])
+            elif line.startswith('- '):
+                differences['removed'].append(line[2:])
+            elif line.startswith('? '):
+                continue
+            else:
+                differences['changed'].append(line[2:])
+        
+        # Generate a summary of differences
+        summary = []
+        if differences['removed']:
+            summary.append("Removed Content:")
+            summary.extend(['• ' + line for line in differences['removed']])
         
-        if response.status_code == 401:
-            return "Unauthorized: Please check your LLaMA API key."
-        elif response.status_code != 200:
-            return f"Error: Received {response.status_code} status code from LLaMA API."
+        if differences['added']:
+            summary.append("\nAdded Content:")
+            summary.extend(['• ' + line for line in differences['added']])
         
-        return response.json()["choices"][0]["text"].strip()
+        if differences['changed']:
+            summary.append("\nModified Content:")
+            summary.extend(['• ' + line for line in differences['changed']])
+        
+        return '\n'.join(summary)
 
 def create_interface():
     assistant = LegalEaseAssistant()
@@ -445,35 +475,66 @@ def create_interface():
                     outputs=risk_output
                 )
 
-            # Legal Assistant Chat Tab
-            with gr.Tab("🤖 Legal Assistant Chat"):
-                chatbot_input = gr.Textbox(
-                    label="💬 Your Message",
-                    placeholder="Ask me anything about legal matters...",
-                    lines=2
+            # Contract Comparison Tab
+            with gr.Tab("📊 Contract Comparison"):
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        contract1_input = gr.File(
+                            label="📎 Upload First Contract",
+                            file_types=[".txt", ".pdf", ".png", ".jpg", ".jpeg", ".tiff", ".bmp"]
+                        )
+                        gr.HTML("<div style='height: 10px'></div>")
+                        contract1_text = gr.Textbox(
+                            label="✍ Or Type/Paste First Contract",
+                            placeholder="Enter your first contract here...",
+                            lines=4
+                        )
+                    
+                    with gr.Column(scale=1):
+                        contract2_input = gr.File(
+                            label="📎 Upload Second Contract",
+                            file_types=[".txt", ".pdf", ".png", ".jpg", ".jpeg", ".tiff", ".bmp"]
+                        )
+                        gr.HTML("<div style='height: 10px'></div>")
+                        contract2_text = gr.Textbox(
+                            label="✍ Or Type/Paste Second Contract",
+                            placeholder="Enter your second contract here...",
+                            lines=4
+                        )
+                
+                compare_btn = gr.Button(
+                    "🔍 Compare Contracts",
+                    variant="primary"
                 )
-                chatbot_output = gr.Textbox(
-                    label="🤖 Assistant Response",
-                    lines=10,
+                
+                comparison_output = gr.Textbox(
+                    label="📊 Comparison Results",
+                    lines=15,
                     show_copy_button=True
                 )
-                chatbot_btn = gr.Button(
-                    "💬 Send Message",
-                    variant="primary"
-                )
-
-                def chatbot_handler(user_input, lang):
-                    if not user_input:
-                        return "Please type a message to start the conversation."
-                    response = assistant.generate_chatbot_response(user_input)
+                
+                def compare_handler(contract1_file, contract1_text, contract2_file, contract2_text, lang):
+                    contract1 = contract1_file or contract1_text
+                    contract2 = contract2_file or contract2_text
+                    
+                    if not contract1 or not contract2:
+                        return "Please provide both contracts for comparison."
+                    
+                    comparison = assistant.compare_contracts(contract1, contract2)
                     if lang != "English":
-                        response = translator.translate_text(response, SUPPORTED_LANGUAGES[lang])
-                    return response
+                        comparison = translator.translate_text(comparison, SUPPORTED_LANGUAGES[lang])
+                    return comparison
                 
-                chatbot_btn.click(
-                    fn=chatbot_handler,
-                    inputs=[chatbot_input, language_selector],
-                    outputs=chatbot_output
+                compare_btn.click(
+                    fn=compare_handler,
+                    inputs=[
+                        contract1_input,
+                        contract1_text,
+                        contract2_input,
+                        contract2_text,
+                        language_selector
+                    ],
+                    outputs=comparison_output
                 )
 
         gr.HTML(f"""