Update app.py

app.py

@@ -32,7 +32,7 @@ class URLProcessor:
         self.session = requests.Session()
         self.timeout = 10  # seconds
         self.session.headers.update({
-            'User -Agent': UserAgent().random,
+            'User-Agent': UserAgent().random,
             'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
             'Accept-Language': 'en-US,en;q=0.5',
             'Accept-Encoding': 'gzip, deflate, br',
@@ -98,7 +98,7 @@ class URLProcessor:
                 return None
                 
             direct_url = f"https://drive.google.com/uc?export=download&id={file_id.group(1)}"
-            response = self.session.get(direct_url, timeout = self.timeout)
+            response = self.session.get(direct_url, timeout=self.timeout)
             response.raise_for_status()
             
             return {
@@ -223,6 +223,32 @@ class FileProcessor:
             logger.error(f"File processing error: {e}")
             return []
 
+class Chatbot:
+    """Simple chatbot that uses provided JSON data for responses."""
+    
+    def __init__(self):
+        self.data = None
+
+    def load_data(self, json_data: str):
+        """Load JSON data into the chatbot."""
+        try:
+            self.data = json.loads(json_data)
+            return "Data loaded successfully!"
+        except json.JSONDecodeError:
+            return "Invalid JSON data. Please check your input."
+
+    def chat(self, user_input: str) -> str:
+        """Generate a response based on user input and loaded data."""
+        if not self.data:
+            return "No data loaded. Please load your JSON data first."
+
+        # Simple keyword-based response logic
+        for key, value in self.data.items():
+            if key.lower() in user_input.lower():
+                return f"{key}: {value}"
+        
+        return "I don't have information on that. Please ask about something else."
+
 def create_interface():
     """Create a comprehensive Gradio interface with advanced features"""
     
@@ -255,11 +281,27 @@ def create_interface():
                 placeholder="Paste your text here..."
             )
         
+        with gr.Tab("Chat"):
+            json_input = gr.Textbox(
+                label="Load JSON Data",
+                placeholder="Paste your JSON data here...",
+                lines=5
+            )
+            load_btn = gr.Button("Load Data", variant="primary")
+            chat_input = gr.Textbox(
+                label="Chat with your data",
+                placeholder="Type your question here..."
+            )
+            chat_output = gr.Textbox(label="Chatbot Response", interactive=False)
+        
         process_btn = gr.Button("Process Input", variant="primary")
         
         output_text = gr.Textbox(label="Processing Results", interactive=False)
         output_file = gr.File(label="Processed Output")
         
+        # Initialize chatbot
+        chatbot = Chatbot()
+
         def process_all_inputs(urls, file, text):
             """Process all input types with progress tracking"""
             try:
@@ -314,18 +356,39 @@ def create_interface():
             except Exception as e:
                 logger.error(f"Processing error: {e}")
                 return None, f"Error: {str(e)}"
-        
+
+        def load_chat_data(json_data):
+            """Load JSON data into the chatbot."""
+            return chatbot.load_data(json_data)
+
+        def chat_with_data(user_input):
+            """Chat with the loaded data."""
+            return chatbot.chat(user_input)
+
         process_btn.click(
             process_all_inputs, 
             inputs=[url_input, file_input, text_input], 
             outputs=[output_file, output_text]
         )
+
+        load_btn.click(
+            load_chat_data,
+            inputs=json_input,
+            outputs=chat_output
+        )
+
+        chat_input.submit(
+            chat_with_data,
+            inputs=chat_input,
+            outputs=chat_output
+        )
         
         gr.Markdown("""
         ### Usage Guidelines
         - **URL Processing**: Enter valid HTTP/HTTPS URLs
         - **File Input**: Upload text files or ZIP archives
         - **Text Input**: Direct text processing
+        - **Chat**: Load JSON data and ask questions about it
         - Advanced cleaning and validation included
         """)