Update app.py

app.py

@@ -134,10 +134,10 @@ def generate_response(message, session_id, model_name, history):
                         response += f"{repo['description']}\n"
                     response += f"⭐ {repo['stargazers_count']} | 🍴 {repo['forks_count']} | Language: {repo['language'] or 'Not specified'}\n"
                     response += f"Updated: {repo['updated_at'][:10]}\n\n"
-                history.append((message, response))
+                history.append({"role": "assistant", "content": response})
                 return history
             else:
-                history.append((message, "No GitHub repositories found for your query."))
+                history.append({"role": "assistant", "content": "No GitHub repositories found for your query."})
                 return history
         
         # Check if it's a Stack Overflow search
@@ -152,10 +152,10 @@ def generate_response(message, session_id, model_name, history):
                     if 'tags' in qa and qa['tags']:
                         response += f"Tags: {', '.join(qa['tags'][:5])}\n"
                     response += f"Asked: {qa['creation_date']}\n\n"
-                history.append((message, response))
+                history.append({"role": "assistant", "content": response})
                 return history
             else:
-                history.append((message, "No Stack Overflow questions found for your query."))
+                history.append({"role": "assistant", "content": "No Stack Overflow questions found for your query."})
                 return history
         
         # Check if it's a code explanation request
@@ -163,7 +163,7 @@ def generate_response(message, session_id, model_name, history):
         if code_match:
             code = code_match.group(1).strip()
             explanation = explain_code(code)
-            history.append((message, explanation))
+            history.append({"role": "assistant", "content": explanation})
             return history
                 
         system_prompt = "You are a technical assistant specializing in software development, programming, and IT topics."
@@ -181,10 +181,10 @@ def generate_response(message, session_id, model_name, history):
             max_tokens=1024
         )
         response = completion.choices[0].message.content
-        history.append((message, response))
+        history.append({"role": "assistant", "content": response})
         return history
     except Exception as e:
-        history.append((message, f"Error generating response: {str(e)}"))
+        history.append({"role": "assistant", "content": f"Error generating response: {str(e)}"})
         return history
 
 # Functions to update PDF viewer
@@ -549,9 +549,18 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     
     with gr.Row(elem_classes="container"):
         with gr.Column(scale=2, min_width=600):
-            chatbot = gr.Chatbot(height=500, bubble_full_width=False, show_copy_button=True, elem_classes="chat-container")
+            chatbot = gr.Chatbot(
+                height=500, 
+                show_copy_button=True, 
+                elem_classes="chat-container",
+                type="messages"
+            )
             with gr.Row():
-                msg = gr.Textbox(show_label=False, placeholder="Ask about your document, type /github to search repos, or /stack to search Stack Overflow...", scale=5)
+                msg = gr.Textbox(
+                    show_label=False, 
+                    placeholder="Ask about your code, type /github to search repos, or /stack to search Stack Overflow...", 
+                    scale=5
+                )
                 send_btn = gr.Button("Send", scale=1)
             clear_btn = gr.Button("Clear Conversation")
     
@@ -592,9 +601,9 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     ).then(lambda: "", None, [msg])
     
     clear_btn.click(
-        lambda: ([], None, "No PDF uploaded yet", {"page_images": [], "total_pages": 0, "total_words": 0}, 0, None, "No PDF uploaded yet"),
+        lambda: ([], None, "No file uploaded", {}, None),
         None,
-        [chatbot, current_session_id, pdf_status, pdf_state, page_slider, pdf_image, stats_display]
+        [chatbot, current_session_id, file_status, code_state, code_metrics]
     )
     
     page_slider.change(