Update app.py

app.py

@@ -11,7 +11,7 @@ import io
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Prompts
+# Prompts remain the same
 CONVERSATION_PROMPT = """You are LOSS DOG, a professional profile builder. Your goal is to have natural conversations
 with users to gather information about their professional background across 9 categories:
 
@@ -167,7 +167,7 @@ class ProfileBuilder:
             
             # Get AI response
             completion = await self.client.chat.completions.create(
-                model="gpt-4-0125-preview",
+                model="gpt-4o-mini",
                 messages=[
                     {"role": "system", "content": CONVERSATION_PROMPT},
                     *conversation_history
@@ -325,7 +325,8 @@ def create_gradio_interface():
                 with gr.Row():
                     msg = gr.Textbox(
                         label="Message",
-                        placeholder="Chat with LOSS DOG..."
+                        placeholder="Chat with LOSS DOG...",
+                        show_label=False
                     )
                     send = gr.Button("Send")
 
@@ -338,15 +339,15 @@ def create_gradio_interface():
         # Event handlers
         async def on_message(message: str, history: List[List[str]], key: str):
             if not message.strip():
-                return history, None, None
+                return history, None, None, ""  # Added empty string to clear input
 
             result = await builder.process_message(message, history, key)
             
             if "error" in result:
-                return history, {"error": result["error"]}, None
+                return history, {"error": result["error"]}, None, message
 
             new_history = history + [[message, result["response"]]]
-            return new_history, None, None
+            return new_history, None, None, ""  # Empty string to clear input
 
         async def on_pdf_upload(pdf, key):
             if not pdf:
@@ -377,13 +378,13 @@ def create_gradio_interface():
         msg.submit(
             on_message,
             inputs=[msg, chatbot, api_key],
-            outputs=[chatbot, profile_output, download_btn]
+            outputs=[chatbot, profile_output, download_btn, msg]  # Added msg to outputs
         )
         
         send.click(
             on_message,
             inputs=[msg, chatbot, api_key],
-            outputs=[chatbot, profile_output, download_btn]
+            outputs=[chatbot, profile_output, download_btn, msg]  # Added msg to outputs
         )
         
         process_pdf_btn.click(