Update app.py

app.py

@@ -1,9 +1,3 @@
-"""
-DOLPHIN PDF Document AI - Local Gemma 3n Version
-Optimized for powerful GPU deployment with local models
-Features: AI-generated alt text for accessibility using local Gemma 3n
-"""
-
 import gradio as gr
 import json
 import markdown
@@ -981,7 +975,7 @@ with gr.Blocks(
             return history
         
         if not processed_markdown:
-            return history + [[message, "❌ Please process a PDF document first before asking questions."]]
+            return history + [{"role": "user", "content": message}, {"role": "assistant", "content": "❌ Please process a PDF document first before asking questions."}]
         
         try:
             # Use RAG to get relevant chunks from markdown
@@ -1009,14 +1003,14 @@ Please provide a clear and helpful answer based on the context provided."""
             
             # Generate response using local Gemma 3n
             response_text = gemma_model.chat(prompt)
-            return history + [[message, response_text]]
+            return history + [{"role": "user", "content": message}, {"role": "assistant", "content": response_text}]
             
         except Exception as e:
             error_msg = f"❌ Error generating response: {str(e)}"
             print(f"Full error: {e}")
             import traceback
             traceback.print_exc()
-            return history + [[message, error_msg]]
+            return history + [{"role": "user", "content": message}, {"role": "assistant", "content": error_msg}]
     
     send_btn.click(
         fn=chatbot_response,