Update app.py

app.py

@@ -21,21 +21,26 @@ def runpod_chat(question, history=[]):
     )
     
     # Stream the response and add to history
-    responses = []
+    full_response = ""
     for message in response_stream:
         response = message.choices[0].delta.content
-        responses.append(response)
+        full_response += "RunPod: " + response + "\n"
         history.append({"role": "assistant", "content": response})
         time.sleep(0.3)  # Simulate typing delay
-        yield "RunPod: " + response
 
-    return history  # Return updated history to maintain state
+    return full_response, history  # Return full response and updated history to maintain state
 
 # Set up the Gradio interface
 iface = gr.Interface(
     fn=runpod_chat,
-    inputs=[gr.Textbox(label="Enter your question:"), gr.State()],
-    outputs=["chat", gr.State()],
+    inputs=[
+        gr.Textbox(label="Enter your question:"),
+        gr.State()
+    ],
+    outputs=[
+        gr.Textbox(label="Responses"),
+        gr.State()
+    ],
     title="RunPod Chat",
     description="This app interfaces with RunPod's API to provide responses to your queries."
 )