Update app.py

app.py

@@ -1,47 +1,49 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Initialize your model
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# Step 1: Set interview type
 def set_type(choice, user_profile):
     user_profile["interview_type"] = choice
     return "Great! What’s your background and what field/role are you aiming for?", user_profile
 
-# Step 2: Save background info
 def save_background(info, user_profile):
     user_profile["field"] = info
     return "Awesome! Type 'start' below to begin your interview.", user_profile
 
-# Step 3: Respond to user
-def respond(message, history, user_profile):
+def respond(message, chat_history, user_profile):
     if not user_profile.get("interview_type") or not user_profile.get("field"):
-        return "Please finish steps 1 and 2 before starting the interview."
+        bot_msg = "Please finish steps 1 and 2 before starting the interview."
+        chat_history.append((message, bot_msg))
+        return chat_history
 
     messages = [
         {"role": "system", "content": f"You are a professional interviewer conducting a {user_profile['interview_type']} interview for a candidate in the {user_profile['field']} field."}
     ]
-    if history:
-        messages.extend(history)
+    for q, a in chat_history:
+        messages.append({"role": "user", "content": q})
+        messages.append({"role": "assistant", "content": a})
     messages.append({"role": "user", "content": message})
 
     response = client.chat_completion(messages, max_tokens=150, stream=False)
-    return response.choices[0].message.content
+    bot_msg = response.choices[0].message.content
+    chat_history.append((message, bot_msg))
+    return chat_history
 
-# Build UI
 with gr.Blocks() as demo:
     user_profile = gr.State({"interview_type": "", "field": ""})
+    chat_history = gr.State([])
 
     gr.Markdown("# 🎤 Welcome to Intervu")
-    
+
     # Step 1
     gr.Markdown("### Step 1: Choose Interview Type")
     with gr.Row():
-        btn1 = gr.Button("Behavioral")
-        btn2 = gr.Button("Technical")
-        btn3 = gr.Button("College / Scholarship")
-    type_output = gr.Textbox(label="Bot", interactive=False)
+        with gr.Column():
+            btn1 = gr.Button("Behavioral")
+            btn2 = gr.Button("Technical")
+            btn3 = gr.Button("College / Scholarship")
+        type_output = gr.Textbox(label="Bot response", interactive=False)
 
     btn1.click(set_type, inputs=[gr.Textbox(value="Behavioral", visible=False), user_profile], outputs=[type_output, user_profile])
     btn2.click(set_type, inputs=[gr.Textbox(value="Technical", visible=False), user_profile], outputs=[type_output, user_profile])
@@ -51,12 +53,17 @@ with gr.Blocks() as demo:
     gr.Markdown("### Step 2: Enter Your Background")
     background = gr.Textbox(label="Your background and field/goal")
     background_btn = gr.Button("Submit")
-    background_output = gr.Textbox(label="Bot", interactive=False)
+    background_output = gr.Textbox(label="Bot response", interactive=False)
 
     background_btn.click(save_background, inputs=[background, user_profile], outputs=[background_output, user_profile])
 
     # Step 3
     gr.Markdown("### Step 3: Start Interview")
-    chatbot = gr.ChatInterface(fn=lambda msg, hist: respond(msg, hist, user_profile.value), title="Intervu - AI Interview Practice", type="messages")
+    chatbot = gr.Chatbot(label="Interview Bot")
+    msg = gr.Textbox(label="Your message")
+    send_btn = gr.Button("Send")
+
+    send_btn.click(respond, inputs=[msg, chat_history, user_profile], outputs=[chatbot], queue=False)
+    send_btn.click(lambda: "", None, msg, queue=False)
 
 demo.launch()