Update app.py

app.py

@@ -1,6 +1,4 @@
 import os
-import json
-from datetime import datetime
 from pathlib import Path
 import gradio as gr
 from huggingface_hub import InferenceClient
@@ -100,7 +98,6 @@ def process_files(files):
     content = "\n\n" + "="*50 + "\n📎 UPLOADED DOCUMENTS\n" + "="*50 + "\n\n"
     
     for file_obj in files:
-        # Handle both file paths and file objects
         file_path = file_obj if isinstance(file_obj, str) else file_obj.name
         file_name = Path(file_path).name
         file_ext = Path(file_path).suffix.lower()
@@ -122,24 +119,41 @@ def process_files(files):
     
     return content
 
-def respond(message, history, system_message, max_tokens, temperature, top_p, model_id, uploaded_files):
-    """Main chat function - Works with ChatInterface"""
-    token = os.getenv("HF_TOKEN")
+def respond(
+    message,
+    history,
+    system_message,
+    max_tokens,
+    temperature,
+    top_p,
+    model_id,
+    uploaded_files,
+    request: gr.Request,  # Gradio 6 OAuth token access
+):
+    """Main chat function - Gradio 6 format with OAuth"""
+    
+    # Get OAuth token from request (Gradio 6 with hf_oauth)
+    token = None
+    if request:
+        token = request.headers.get("authorization", "").replace("Bearer ", "")
+    
+    # Fallback to environment variable
+    if not token:
+        token = os.getenv("HF_TOKEN")
     
     if not token:
-        yield "⚠️ HF_TOKEN not configured. Please set it in Space settings → Repository secrets."
+        yield "⚠️ Please log in with your Hugging Face account to use this Space."
         return
     
     try:
         client = InferenceClient(token=token, model=model_id)
         
-        # Build messages
+        # Build messages in OpenAI format (required for Gradio 6)
         messages = [{"role": "system", "content": system_message}]
         
-        # Add history
-        for user_msg, assistant_msg in history:
-            messages.append({"role": "user", "content": user_msg})
-            messages.append({"role": "assistant", "content": assistant_msg})
+        # Add history - Gradio 6 uses messages format
+        for msg in history:
+            messages.append(msg)
         
         # Process uploaded files
         file_content = ""
@@ -164,34 +178,41 @@ def respond(message, history, system_message, max_tokens, temperature, top_p, mo
                 yield response
                 
     except Exception as e:
-        yield f"❌ Error: {str(e)}\n\nTry a different model or check your token."
+        yield f"❌ Error: {str(e)}\n\nTry a different model or refresh the page."
 
 def update_system_prompt(preset):
     return SYSTEM_PROMPTS.get(preset, SYSTEM_PROMPTS["Default"])
 
-# Build interface using ChatInterface
-with gr.Blocks(title="AI Document Assistant") as demo:
+# Gradio 6 Interface
+with gr.Blocks(title="Jarvis - AI Document Assistant") as demo:
     
     gr.Markdown(
         """
-        # 🤖 AI Document Assistant
-        Upload documents (PDF, DOCX, TXT, CSV) and chat with AI about their content
+        # 💬 Jarvis - AI  Assistant
+        Upload documents (PDF, DOCX, TXT, CSV) and chat with powerful AI models
         """
     )
     
     with gr.Row():
         with gr.Column(scale=3):
-            # File upload area
+            # File upload
             file_upload = gr.File(
                 label="📁 Upload Documents",
                 file_count="multiple",
                 file_types=[".pdf", ".docx", ".txt", ".csv", ".md", ".py", ".json"],
             )
             
-            # ChatInterface handles all the message formatting automatically
+            # ChatInterface - Gradio 6 format
             chat = gr.ChatInterface(
                 fn=respond,
-                chatbot=gr.Chatbot(height=500, show_label=False),
+                chatbot=gr.Chatbot(
+                    height=500,
+                    show_label=False,
+                    avatar_images=(
+                        None,
+                        "https://huggingface.co/datasets/huggingface/brand-assets/resolve/main/hf-logo.png"
+                    ),
+                ),
                 textbox=gr.Textbox(
                     placeholder="💬 Ask a question about your documents...",
                     show_label=False,
@@ -199,15 +220,14 @@ with gr.Blocks(title="AI Document Assistant") as demo:
                 additional_inputs=[
                     gr.Textbox(
                         value=SYSTEM_PROMPTS["Document Analyzer"],
-                        label="💬 System Prompt",
-                        lines=3,
-                        visible=False,  # Hidden in settings panel
+                        label="System Prompt",
+                        visible=False,
                     ),
-                    gr.Slider(128, 4096, 2048, step=128, label="Max Tokens", visible=False),
-                    gr.Slider(0.1, 2.0, 0.7, step=0.1, label="Temperature", visible=False),
-                    gr.Slider(0.1, 1.0, 0.95, step=0.05, label="Top-p", visible=False),
-                    gr.Dropdown(choices=MODELS, value=MODELS[0], label="Model", visible=False),
-                    file_upload,  # Pass file upload to the function
+                    gr.Slider(128, 4096, 2048, step=128, visible=False),
+                    gr.Slider(0.1, 2.0, 0.7, step=0.1, visible=False),
+                    gr.Slider(0.1, 1.0, 0.95, step=0.05, visible=False),
+                    gr.Dropdown(choices=MODELS, value=MODELS[0], visible=False),
+                    file_upload,
                 ],
                 retry_btn="🔄 Retry",
                 undo_btn="↩️ Undo",
@@ -220,13 +240,13 @@ with gr.Blocks(title="AI Document Assistant") as demo:
             model_dropdown = gr.Dropdown(
                 choices=MODELS,
                 value=MODELS[0],
-                label="🤖 Model",
+                label="🤖 AI Model",
             )
             
             preset_dropdown = gr.Dropdown(
                 choices=list(SYSTEM_PROMPTS.keys()),
                 value="Document Analyzer",
-                label="📋 Preset",
+                label="📋 Behavior Preset",
             )
             
             system_prompt = gr.Textbox(
@@ -235,17 +255,34 @@ with gr.Blocks(title="AI Document Assistant") as demo:
                 lines=4,
             )
             
-            gr.Markdown("### 🎛️ Parameters")
+            gr.Markdown("### 🎛️ Generation")
             
-            max_tokens = gr.Slider(128, 4096, 2048, step=128, label="Max Tokens")
-            temperature = gr.Slider(0.1, 2.0, 0.7, step=0.1, label="Temperature")
-            top_p = gr.Slider(0.1, 1.0, 0.95, step=0.05, label="Top-p")
+            max_tokens = gr.Slider(
+                128, 4096, 2048, 
+                step=128, 
+                label="Max Tokens",
+                info="Maximum response length"
+            )
+            
+            temperature = gr.Slider(
+                0.1, 2.0, 0.7, 
+                step=0.1, 
+                label="Temperature",
+                info="Creativity (higher = more random)"
+            )
+            
+            top_p = gr.Slider(
+                0.1, 1.0, 0.95, 
+                step=0.05, 
+                label="Top-p",
+                info="Nucleus sampling"
+            )
     
-    # Link settings to ChatInterface hidden inputs
+    # Connect settings to ChatInterface
     model_dropdown.change(
         lambda x: x,
         inputs=[model_dropdown],
-        outputs=[chat.additional_inputs[4]],  # Update model in ChatInterface
+        outputs=[chat.additional_inputs[4]],
     )
     
     preset_dropdown.change(
@@ -257,7 +294,7 @@ with gr.Blocks(title="AI Document Assistant") as demo:
     system_prompt.change(
         lambda x: x,
         inputs=[system_prompt],
-        outputs=[chat.additional_inputs[0]],  # Update system prompt in ChatInterface
+        outputs=[chat.additional_inputs[0]],
     )
     
     max_tokens.change(
@@ -281,14 +318,24 @@ with gr.Blocks(title="AI Document Assistant") as demo:
     gr.Markdown(
         """
         ---
-        **💡 How to use:**
-        1. Upload your documents (PDF, DOCX, TXT, CSV) using the file uploader
-        2. Ask questions about the content in the chat
-        3. Adjust settings on the right panel
-        4. Lower temperature = focused responses, Higher = creative
+        ### 💡 How to Use
+        
+        1. **Log in** with your Hugging Face account (OAuth enabled)
+        2. **Upload documents** - PDF, DOCX, TXT, CSV supported
+        3. **Ask questions** about the content
+        4. **Adjust settings** for different response styles
+        
+        ### 📊 Supported Formats
+        - **PDF**: Text extraction from all pages
+        - **DOCX**: Microsoft Word documents
+        - **TXT/MD**: Plain text and Markdown
+        - **CSV**: Data files with statistics
+        - **Code**: Python, JavaScript, JSON, etc.
         
-        **📦 Requires:** pypdf, python-docx, pandas
-        **🔐 Setup:** Add HF_TOKEN in Space settings → Repository secrets
+        ### 🎯 Tips
+        - Lower temperature (0.1-0.5) = Focused, deterministic
+        - Higher temperature (0.8-2.0) = Creative, varied
+        - Try different models for different tasks
         """
     )