Update app2.py

app2.py

@@ -3,6 +3,8 @@ import os
 import re
 import logging
 import mimetypes
+import time
+import io
 from selenium import webdriver
 from chromedriver_py import binary_path
 import concurrent.futures
@@ -21,6 +23,8 @@ from fake_useragent import UserAgent
 from ratelimit import limits, sleep_and_retry
 from cleantext import clean
 import qrcode
+from PIL import Image
+from pyzbar.pyzbar import decode
 import nest_asyncio
 nest_asyncio.apply()
 import aiohttp
@@ -225,7 +229,7 @@ class FileProcessor:
                                     "source": "file",
                                     "filename": filename,
                                     "content": content,
-                                    "timestamp": datetime.now().isoformat()
+                                    "timestamp": datetime.now ().isoformat()
                                 })
                         except Exception as e:
                             logger.error(f"Error reading file {filename}: {str(e)}")
@@ -347,9 +351,54 @@ def generate_qr_code(data: Union[str, Dict], combined: bool = True) -> List[str]
         logger.error(f"QR generation error: {e}")
         return []
 
+def decode_qr_code(image_path: str) -> str:
+    """Decode QR code from an image file"""
+    try:
+        img = Image.open(image_path)
+        decoded_objects = decode(img if decoded_objects:
+            return decoded_objects[0].data.decode('utf-8')
+        raise ValueError("Unable to decode QR code")
+    except Exception as e:
+        logger.error(f"QR decoding error: {e}")
+        return None
+
+def datachat_trained(data_input: str, query: str) -> str:
+    """Handle trained data interaction logic"""
+    data = clean_json(data_input)
+    if not data:
+        return "Invalid JSON data provided."
+    return f"[Trained Mode]\nData: {json.dumps(data, indent=2)}\nQuery: {query}"
+
+def datachat_simple(data_input: str, query: str) -> str:
+    """Handle simple chat interaction logic"""
+    data = clean_json(data_input)
+    if not data:
+        return "Invalid JSON data provided."
+    return f"[Chat Mode]\nData: {json.dumps(data, indent=2)}\nQuestion: {query}"
+
+def datachat_interface(mode: str, data_source: str, json_input: str, qr_image: str, query: str) -> str:
+    """Interface for DataChat functionality"""
+    data = None
+    if data_source == "JSON Input":
+        data = json_input
+    elif data_source == "QR Code":
+        try:
+            decoded_data = decode_qr_code(qr_image)
+            data = decoded_data
+        except Exception as e:
+            return f"Invalid QR code data provided: {e}"
+    else:
+        return "No valid data source selected."
+
+    if mode == "Trained with Data":
+        return datachat_trained(data, query)
+    elif mode == "Chat about Data":
+        return datachat_simple(data, query)
+    else:
+        return "Invalid mode selected."
+
 def create_interface():
     """Create a comprehensive Gradio interface with advanced features"""
-    
     css = """
     .container { max-width: 1200px; margin: auto; }
     .warning { background-color: #fff3cd; color: #856404; padding: 10px; border-radius: 4px; }
@@ -357,158 +406,34 @@ def create_interface():
     .success { background-color: #d4edda; color: #155724; padding: 10px; border-radius: 4px; }
     """
     
-    with gr.Blocks(css=css, title="Advanced Data Processor & QR Generator") as interface:
+    with gr.Blocks(css=css, title="Advanced Data Processor & QR Code Generator") as interface:
         gr.Markdown("# 🌐 Advanced Data Processing & QR Code Generator")
         
-        with gr.Tab("URL Processing"):
-            url_input = gr.Textbox(
-                label="Enter URLs (comma or newline separated)", 
-                lines=5,
-                placeholder="https://example1.com\nhttps://example2.com",
-                value=""
-            )
-        
-        with gr.Tab("File Input"):
-            file_input = gr.File(
-                label="Upload text file or ZIP archive",
-                file_types=[".txt", ".zip", ".md", ".csv", ".json", ".xml"]
-            )
-
-        with gr.Tab("Notepad"):
-            text_input = gr.TextArea(
-                label="JSON Data Input", 
-                lines=15,
-                placeholder="Paste your JSON data here...",
-                value=""
-            )
-            
-            with gr.Row():
-                example_btn = gr.Button("📝 Load Example JSON", variant="secondary")
-                clear_btn = gr.Button("🗑️ Clear Input", variant="secondary")
-            
-        with gr.Row():
-            combine_data = gr.Checkbox(
-                label="Combine all data into single QR code", 
-                value=True,
-                info="Generate one QR code for all data, or separate QR codes for each item"
-            )
-            process_btn = gr.Button("🔄 Process & Generate QR", variant="primary", scale=2)
-        
-        output_json = gr.JSON(label="Processed JSON Data")
-        output_gallery = gr.Gallery(label="Generated QR Codes", columns=2, height=400)
-        output_text = gr.Textbox(label="Processing Status", interactive=False)
-        
-        def load_example():
-            example_json = {
-                "type": "product_catalog",
-                "items": [
-                    {
-                        "id": "123",
-                        "name": "Test Product",
-                        "description": "This is a test product description",
-                        "price": 29.99,
-                        "category": "electronics",
-                        "tags": ["test", "sample", "demo"]
-                    },
-                    {
-                        "id": "456",
-                        "name": "Another Product",
-                        "description": "Another test product description",
-                        "price": 49.99,
-                        "category": "accessories",
-                        "tags": ["sample", "test"]
-                    }
-                ],
-                "metadata": {
-                    "timestamp": datetime.now().isoformat(),
-                    "version": "1.0",
-                    "source": "example"
-                }
-            }
-            return json.dumps(example_json, indent=2)
-        
-        def clear_input():
-            return ""
-        
-        def process_all_inputs(urls, file, text, combine):
-            """Process all input types and generate QR codes"""
-            try:
-                results = []
-                
-                if text and text.strip():
-                    try:
-                        json_data = json.loads(text)
-                        if isinstance(json_data, list):
-                            results.extend(json_data)
-                        else:
-                            results.append(json_data)
-                    except json.JSONDecodeError as e:
-                        return None, [], f"❌ Invalid JSON format: {str(e)}"
-                
-                if urls and urls.strip():
-                    processor = URLProcessor()
-                    url_list = re.split(r'[,\n]', urls)
-                    url_list = [url.strip() for url in url_list if url.strip()]
-                    
-                    for url in url_list:
-                        validation = processor.validate_url(url)
-                        if validation.get('is_valid'):
-                            content = processor.fetch_content(url)
-                            if content:
-                                results.append({
-                                    'source': 'url',
-                                    'url': url,
-                                    'content': content,
-                                    'timestamp': datetime.now().isoformat()
-                                })
-                
-                if file:
-                    file_processor = FileProcessor()
-                    file_results = file_processor.process_file(file)
-                    if file_results:
-                        results.extend(file_results)
-                
-                if results:
-                    qr_paths = generate_qr_code(results, combined=combine)
-                    if qr_paths:
-                        return (
-                            results,
-                            [str(path) for path in qr_paths],
-                            f"✅ Successfully processed {len(results)} items and generated {len(qr_paths)} QR code(s)!"
-                        )
-                    else:
-                        return None, [], "❌ Failed to generate QR codes. Please check the input data."
-                else:
-                    return None, [], "⚠️ No valid content to process. Please provide some input data."
-            
-            except Exception as e:
-                logger.error(f"Processing error: {e}")
-                return None, [], f"❌ Error: {str(e)}"
-        
-        example_btn.click(load_example, outputs=[text_input])
-        clear_btn.click(clear_input, outputs=[text_input])
-        process_btn.click(
-            process_all_inputs, 
-            inputs=[url_input, file_input, text_input, combine_data], 
-            outputs=[output_json, output_gallery, output_text]
-        )
-        
-        gr.Markdown("""
-        ### Features
-        - **URL Processing**: Extract content from websites
-        - **File Processing**: Handle text files and archives
-        - **Notepad**: Direct JSON data input/manipulation
-        - **JSON Cleaning**: Automatic JSON validation and formatting
-        - **QR Generation**: Generate QR codes with embedded JSON data
-        - **Flexible Output**: Choose between combined or separate QR codes
-        
-        ### Usage Tips
-        1. Use the **Notepad** tab for direct JSON input
-        2. Click "Load Example JSON" to see a sample format
-        3. Choose whether to combine all data into a single QR code
-        4. The generated QR codes will contain the complete JSON data
-        """)
-    
+        with gr.Tab("DataChat"):
+            mode = gr.Radio(["Trained with Data", "Chat about Data"], label="Mode")
+            data_source = gr.Radio(["JSON Input", "QR Code"], label="Data Source")
+            json_input = gr.Textbox(lines=8, label="JSON Data")
+            qr_image = gr.Image(label="QR Code Image", type="filepath")
+            query = gr.Textbox(label="Query")
+
+            submit_btn = gr.Button("Submit")
+            output = gr.Textbox(label="Response")
+
+            submit_btn.click(datachat_interface, [mode, data_source, json_input, qr_image, query], output)
+
+        with gr.Tab("QR Generator"):
+            qr_input = gr.Textbox(lines=8, label="Input JSON for QR")
+            generate_btn = gr.Button("Generate QR")
+            qr_output = gr.Image(label="Generated QR Code")
+
+            def generate_qr(json_data):
+                data = clean_json(json_data)
+                if data:
+                    return generate_qr_code(data)
+                return None
+
+            generate_btn.click(generate_qr, qr_input, qr_output)
+
     return interface
 
 def main():