Update app.py

app.py

@@ -58,8 +58,7 @@ def get_streamed_response(message, history):
 
             # Check for the completion signal
             if decoded_line == "data: [DONE]":
-                yield entire_assistant_response  # Yield the entire response at the end
-                break
+                return entire_assistant_response  # Return the entire response at the end
 
             try:
                 # Decode and strip any SSE format specific prefix ("data: ")
@@ -68,7 +67,6 @@ def get_streamed_response(message, history):
                     chunk_data = json.loads(decoded_line)
                     content = chunk_data['choices'][0]['delta']['content']
                     entire_assistant_response += content  # Aggregate content
-                    yield entire_assistant_response
 
             except json.JSONDecodeError:
                 print(f"Invalid JSON received: {decoded_line}")
@@ -80,13 +78,17 @@ def get_streamed_response(message, history):
     print(entire_assistant_response)
     all_message.append({"role": "assistant", "content": entire_assistant_response})
 
+# Function to update the output textbox dynamically
+def update_output(message):
+    return message
+
 # Create a Gradio interface with customizations
 chat_interface = gr.Interface(
     fn=get_streamed_response,
+    inputs=[gr.Textbox(label="You", lines=5, placeholder="Type your message here...")],
+    outputs=update_output,
     title="Raxder Official AI",
     description="Welcome to Dave, your friendly AI Therapist! Ask me anything or just have a chat. 😊",
-    inputs=[gr.Textbox(label="You", lines=5, placeholder="Type your message here...")],
-    outputs=[gr.Textbox(label="Dave", lines=5, placeholder="Dave's response will appear here...", readonly=True)],
     theme="huggingface",
     layout="vertical",
     allow_flagging=False,