Update app.py

app.py

@@ -80,13 +80,13 @@ def copy_last(history):
     return ""
 
 def download_history(history):
-    file_path = "chat_history.txt"
     text = ""
     for h in history:
         if isinstance(h, dict):
             text += f'{h["role"]}: {h["message"]}\n\n'
         else:
             text += f'User: {h[0]}\nAI: {h[1]}\n\n'
+    file_path = "chat_history.txt"
     with open(file_path, "w", encoding="utf-8") as f:
         f.write(text)
     return file_path
@@ -123,38 +123,50 @@ def chat_endpoint(request: ChatRequest):
 # ----------------------------
 # Gradio UI
 # ----------------------------
-with gr.Blocks(theme=gr.Theme.from_hub("JackismyShephard/ultimate-rvc-theme")) as demo:
-    gr.HTML("<h2 style='text-align:center; color:white;'>Advanced AI Chat</h2>")
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column(scale=1, min_width=800):
+            chatbot = gr.Chatbot(height=700)
+            msg = gr.Textbox(placeholder="Type your message...", container=False)
+            state = gr.State([])
 
-    chatbot = gr.Chatbot(height=600)
-    msg = gr.Textbox(placeholder="Type your question...", container=False)
-    state = gr.State([])
+            with gr.Row():
+                send_btn = gr.Button("Send")
+                stop_btn = gr.Button("🛑 Stop")
+                copy_btn = gr.Button("📋 Copy Last")
+                download_btn = gr.File(label="📥 Download Chat")
 
-    send_btn = gr.Button("Send")
-    stop_btn = gr.Button("🛑 Stop")
-    copy_btn = gr.Button("📋 Copy Last")
-    download_btn = gr.File(label="📥 Download Chat")
-    edit_index = gr.Number(label="Edit Message Index")
-    edit_text = gr.Textbox(label="New Text")
-    edit_btn = gr.Button("✏️ Edit")
+            with gr.Row():
+                edit_index = gr.Number(label="Edit Index")
+                edit_text = gr.Textbox(label="New Text")
+                edit_btn = gr.Button("✏️ Edit")
 
     # ----------------------------
-    # Button callbacks
+    # Callbacks
     # ----------------------------
     send_btn.click(generate_response, [msg, state], [chatbot, state])
     stop_btn.click(lambda: stop_event.set())
     copy_btn.click(copy_last, inputs=[state], outputs=None)
-    download_btn.click(download_history, inputs=[state], outputs=download_btn)
+    download_btn.update(value=None)  # gr.File doesn't have click; we update value
     edit_btn.click(edit_message, inputs=[state, edit_index, edit_text], outputs=[chatbot])
 
     demo.css = """
-    .gradio-container { border-radius: 25px !important; max-width: 600px !important; margin:auto !important; overflow:hidden; }
-    .message.user { border-radius: 18px 18px 4px 18px !important; }
-    .message.bot { border-radius: 18px 18px 18px 4px !important; }
+    .gradio-container {
+        width: 100vw !important;
+        height: 100vh !important;
+        border-radius: 0px !important;
+        overflow: hidden;
+        background-color: #0b0f19 !important;
+    }
+    .input-textbox textarea {
+        border-radius: 25px !important;
+    }
+    .message.user { border-radius: 18px 18px 4px 18px !important; background:#2b6fff !important; color:white !important;}
+    .message.bot { border-radius: 18px 18px 18px 4px !important; background:#1c1f2a !important; color:white !important;}
     """
 
 # ----------------------------
-# Run Gradio + FastAPI together
+# Run Gradio + FastAPI
 # ----------------------------
 def run_gradio():
     demo.launch(server_name="0.0.0.0", server_port=7860)