Update app.py

app.py

@@ -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 {
@@ -156,6 +156,15 @@ class FileProcessor:
         self.max_file_size = max_file_size
         self.supported_text_extensions = {'.txt', '.md', '.csv', '.json', '.xml'}
 
+    def is_text_file(self, filepath: str) -> bool:
+        """Check if file is a text file"""
+        try:
+            mime_type, _ = mimetypes.guess_type(filepath)
+            return (mime_type and mime_type.startswith('text/')) or \
+                   (os.path.splitext(filepath)[1].lower() in self.supported_text_extensions)
+        except Exception:
+            return False
+
     def process_file(self, file) -> List[Dict]:
         """Process uploaded file with enhanced error handling"""
         if not file:
@@ -223,6 +232,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 +290,27 @@ def create_interface():
                 placeholder="Paste your text here..."
             )
         
+        with gr.Tab("Chat"):
+            chat_input = gr.Textbox(
+                label="Chat with your data",
+                placeholder="Type your question here..."
+            )
+            json_input = gr.Textbox(
+                label="Load JSON Data",
+                placeholder="Paste your JSON data here...",
+                lines=5
+            )
+            load_btn = gr.Button("Load Data")
+            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 +365,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 your JSON data and ask questions about it
         - Advanced cleaning and validation included
         """)
     
@@ -342,9 +414,9 @@ def main():
     interface.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=False,
-        inbrowser=False,  # Disable browser opening in container
-        debug=False       # Disable debug mode for production
+        share=True,  # Enable public sharing
+        inbrowser=False,
+        debug=False
     )
 
 if __name__ == "__main__":