Update app.py

app.py

@@ -1,7 +1,45 @@
+import os
 import gradio as gr
+from gradio import blocks
 
-def greet(name: gr.inputs.Text()):
-    return "Hello, " + name + "!"
+model = "OpenAssistant/oasst-sft-1-pythia-12b"
+client = InferenceAPIClient(model)
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+# creating a list to store the chat history
+history = []
+
+def generate_response(inputs):
+    # input validation
+    if not inputs:
+        return "Please enter a valid input."
+
+    # checking if the input is a continuation of a conversation or a new conversation
+    if history and history[-1]["speaker"] == "user":
+        inputs = history[-1]["text"] + " " + inputs
+
+    # generating a response using the inference api client
+    response = client.generate_utterance(inputs)
+
+    # adding the user input and the model's response to the history list
+    history.append({"speaker": "user", "text": inputs})
+    history.append({"speaker": "model", "text": response})
+
+    return response
+
+iface = gr.Interface(
+    generate_response,
+    [
+        gr.inputs.Textbox(
+            placeholder="Hi, how can I help you?",
+            label="User Input"
+        )
+    ],
+    [gr.outputs.Textbox(label="Model Response")],
+    title="OpenAssistant: AI Powered Chatbot",
+    live=True,
+    layout="vertical",
+    theme="compact"
+)
+
+if __name__ == '__main__':
+    iface.launch()