Update app.py

app.py

@@ -27,13 +27,21 @@ def count_tokens(text: str, model: str) -> int:
     enc = _get_encoding(model)
     return len(enc.encode(text))
 
-# Read uploaded file
+# Read uploaded file content (filepath or file-like)
 def read_file_content(file_obj):
     if not file_obj:
         return ""
     try:
-        content = file_obj.read().decode('utf-8')
-        return f"\n\n--- Start of file: {file_obj.name} ---\n{content}\n--- End of file ---\n"
+        # If filepath, open and read
+        if isinstance(file_obj, str):
+            with open(file_obj, 'r', encoding='utf-8') as f:
+                content = f.read()
+            name = os.path.basename(file_obj)
+        else:
+            # file_obj is a file-like object
+            name = getattr(file_obj, 'name', 'uploaded_file')
+            content = file_obj.read().decode('utf-8')
+        return f"\n\n--- Start of file: {name} ---\n{content}\n--- End of file ---\n"
     except Exception:
         return ""
 
@@ -57,18 +65,22 @@ def respond(message, history, model_name, file_obj):
     completion_tokens = count_tokens(reply, model_name)
     usage_info = f"(Tokens used: prompt={prompt_tokens}, completion={completion_tokens})"
 
-    # Update history
-    history.append((message, f"{reply}\n\n{usage_info}"))
+    # Update history (OpenAI-style messages)
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": f"{reply}\n\n{usage_info}"})
     return history
 
 # Build Gradio interface
 with gr.Blocks() as demo:
     model_dropdown = gr.Dropdown(ALL_MODELS, value=ALL_MODELS[0], label="Select Model")
-    chatbot = gr.Chatbot(label="Chat with AI")
-    file_upload = gr.File(label="Upload File", file_types=[".txt", ".md", ".py"], type="file")
+    chatbot = gr.Chatbot(label="Chat with AI", type="messages")
+    file_upload = gr.File(label="Upload File", file_types=[".txt", ".md", ".py"], type="filepath")
     user_input = gr.Textbox(placeholder="Type your message...", show_label=False)
 
     # Link interactions
-    user_input.submit(respond, inputs=[user_input, chatbot, model_dropdown, file_upload], outputs=chatbot)
+    user_input.submit(respond,
+                      inputs=[user_input, chatbot, model_dropdown, file_upload],
+                      outputs=chatbot)
     user_input.submit(lambda: "", None, user_input)
+
     demo.launch()