Update app.py

app.py

@@ -19,41 +19,30 @@ def process_file(filepath):
     if not filepath:
         return ""
     
-    # Get file extension
-    # e.g., if filepath = "/somefolder/report.pdf", ext = ".pdf"
     ext = os.path.splitext(filepath)[1].lower()
     
     try:
         if ext in ['.png', '.jpg', '.jpeg']:
-            # OCR processing for images
             image = Image.open(filepath)
             text = pytesseract.image_to_string(image)
             return f"IMAGE CONTENT:\n{text}"
-            
         elif ext == '.pdf':
-            # PDF text extraction
             pdf_reader = PyPDF2.PdfReader(filepath)
-            text = "\n".join([page.extract_text() for page in pdf_reader.pages])
+            text = "\n".join(page.extract_text() for page in pdf_reader.pages)
             return f"PDF CONTENT:\n{text}"
-            
         elif ext == '.docx':
-            # Word document processing
             doc = docx.Document(filepath)
-            text = "\n".join([para.text for para in doc.paragraphs])
+            text = "\n".join(para.text for para in doc.paragraphs)
             return f"DOCUMENT CONTENT:\n{text}"
-            
         else:
             return "Unsupported file type"
-            
     except Exception as e:
         print(f"File processing error: {e}")
         return "Error reading file"
 
 def chat(message, history, filepath):
-    # Process uploaded file
     file_content = process_file(filepath) if filepath else ""
     
-    # Build enhanced prompt
     full_prompt = f"""
     {file_content}
     
@@ -61,8 +50,7 @@ def chat(message, history, filepath):
     
     Please respond considering both the message and any attached documents:
     """
-    
-    # Configure generation parameters
+
     generate_kwargs = dict(
         temperature=0.7,
         max_new_tokens=2000,
@@ -80,30 +68,25 @@ def chat(message, history, filepath):
     
     partial_message = ""
     for response in stream:
-        # Skip special tokens
         if response.token.special:
             continue
         partial_message += response.token.text
         yield partial_message
 
-# Create Gradio interface with file upload
 with gr.Blocks(theme="soft") as demo:
     gr.Markdown("# DeepSeek-R1 Assistant with File Support")
     gr.Markdown("Upload images, PDFs, or docs and chat about them!")
     
     with gr.Row():
-        # Note the change: type="filepath"
         file_input = gr.File(label="Upload File (PDF/Image/Doc)", type="filepath")
         
-    # The ChatInterface should pass the file path to `chat`
     chatbot = gr.ChatInterface(
         fn=chat,
         additional_inputs=[file_input],
-        examples=[
-            ["Explain this document", "report.pdf"],
-            ["What's in this image?", "screenshot.png"]
-        ]
+        type="messages",
+        examples=[]
     )
 
 demo.launch()
 
+