Update app.py

app.py

@@ -1,15 +1,8 @@
-import gradio as gr
 import streamlit as st
-import requests
 from gradio_client import Client
 
-ngrok_tunnel_url = "https://ngrok-python/1.4.0 ngrok-rust/0.14.0-pre.14L/predict"
-
-
-
-st.title('Digital Ink')
-
-
+#connecting to the URL
+client = Client("https://f77c234af73e4f6a6f.gradio.live/")
 
 # Initialize chat history
 if "messages" not in st.session_state:
@@ -20,17 +13,19 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
-def get_response(prompt):
-    response = requests.post(f"{ngrok_tunnel_url}/predict", json={"text": prompt})
-    return response.json()["output"]
+# Function for generating responses
+def generate_response(message, model, tokenizer):
+    st.session_state.messages.append({"role": "user", "content": message})
+    response = client.predict(message=message,api_name="/predict")
+    st.session_state.messages.append({"role": "assistant", "content": response})
     return response
 
-# User input
-if prompt := st.chat_input("Ask Digital Ink..."):
+# User input with placeholder
+if prompt := st.chat_input("Create with Digital Ink.."):
     # Display user message in chat message container
     with st.chat_message("user"):
         st.markdown(prompt)
     # Generate and display assistant response in chat message container
     with st.chat_message("assistant"):
-        response = get_response(prompt)
+        response = generate_response(prompt)
         st.markdown(response)