Update app.py

app.py

@@ -1,51 +1,31 @@
-import requests
-from langchain import PromptTemplate
-
-# Define a simple prompt template
-prompt_template = PromptTemplate(
-    input_variables=['name'],
-    template='Hello {name}, how can I assist you today?'
-)
-
-# Define a function to interact with Google Gemini API
-def generate_response(name):
-    # Construct the prompt
-    prompt = prompt_template.format(name=name)
-    
-    # Define API endpoint and headers
-    api_url = 'https://your-google-gemini-endpoint.com/v1/generate'
-    headers = {
-        'Authorization': f'Bearer YOUR_GEMINI_API_KEY',
-        'Content-Type': 'application/json'
-    }
-    payload = {
-        'prompt': prompt
-    }
-
-    # Send the request to the API
-    try:
-        print(f'Sending request to: {api_url}')
-        response = requests.post(api_url, headers=headers, json=payload)
-        print(f'Response status code: {response.status_code}')
-        if response.status_code == 200:
-            result = response.json()
-            generated_text = result.get('text', 'No response text available.')
+import streamlit as st
+from langsmith import LangsmithClient
+
+# Initialize the Langsmith client with your API key
+api_key = st.secrets["LANGSMITH_API_KEY"]
+client = LangsmithClient(api_key)
+
+st.title("Custom Customer Support Bot")
+
+def generate_response(query):
+    response = client.chat(
+        model="text-davinci-003",  # You can choose an appropriate model
+        prompt=f"Customer query: {query}\n\nResponse:",
+        max_tokens=150
+    )
+    return response['choices'][0]['text'].strip()
+
+with st.form("support_form"):
+    user_query = st.text_area("Enter your query here:")
+    submitted = st.form_submit_button("Submit")
+
+    if submitted:
+        if user_query:
+            response = generate_response(user_query)
+            st.write("**Bot Response:**")
+            st.write(response)
         else:
-            generated_text = f'Error: Unexpected status code {response.status_code}'
-    except requests.exceptions.RequestException as e:
-        generated_text = f'Error: {e}'
-
-    print(f'Tracking: LLM Request - Name: {name}, Prompt: {prompt}')
-    print(f'Tracking: LLM Response - Response: {generated_text}')
-
-    return generated_text
-
-# Example usage
-if __name__ == "__main__":
-    name = "Alice"
-    response = generate_response(name)
-    print(f"Generated Response: {response}")
-
+            st.error("Please enter a query to get a response.")