Update app.py

app.py

@@ -1,34 +1,45 @@
-import os
 import gradio as gr
-from langchain.chat_models import ChatOpenAI
-from langchain import LLMChain, PromptTemplate
-from langchain.memory import ConversationBufferMemory
+from groq import Groq
 
-OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
+client = Groq(api_key='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')
 
-template = """You are a helpful assistant to answer all user queries.
-{chat_history}
-User: {user_message}
-Chatbot:"""
+def chatbot(input_text, chat_history):
+   
+    messages = [{"role": "system", "content": "answer to every input as if you are a tech-savvy computer science student who spends countless hours coding, building apps, and keeping up with the latest tech trends. You enjoy discussing programming languages, AI, and gadgets and are always ready to troubleshoot tech-related problems."}]
+    for history in chat_history:
+        messages.append({"role": "user", "content": history[0]})
+        messages.append({"role": "assistant", "content": history[1]})
 
-prompt = PromptTemplate(
-    input_variables=["chat_history", "user_message"], template=template
-)
+    messages.append({"role": "user", "content": input_text})
 
-memory = ConversationBufferMemory(memory_key="chat_history")
+    completion = client.chat.completions.create(
+        model="llama3-8b-8192",
+        messages=messages,
+        temperature=1,
+        max_tokens=1024,
+        top_p=1,
+        stream=False, 
+        stop=None,
+    )
 
-llm_chain = LLMChain(
-    llm=ChatOpenAI(temperature='0.5', model_name="gpt-3.5-turbo"),
-    prompt=prompt,
-    verbose=True,
-    memory=memory,
-)
+    assistant_reply = completion.choices[0].message.content 
+    
+    chat_history.append((input_text, assistant_reply))
 
-def get_text_response(user_message,history):
-    response = llm_chain.predict(user_message = user_message)
-    return response
+    return chat_history, chat_history 
 
-demo = gr.ChatInterface(get_text_response)
+
+with gr.Blocks() as demo:
+    gr.Markdown("## Chat with Tech-Savvy Bot")
+    
+    chatbot_interface = gr.Chatbot()
+    msg = gr.Textbox(placeholder="Type your message here...")
+    clear_btn = gr.Button("Clear Chat")
+    
+    chat_history = gr.State([]) 
+
+    msg.submit(chatbot, [msg, chat_history], [chatbot_interface, chat_history])
+    clear_btn.click(lambda: ([], []), None, [chatbot_interface, chat_history])
 
 if __name__ == "__main__":
-    demo.launch() #To create a public link, set `share=True` in `launch()`. To enable errors and logs, set `debug=True` in `launch()`.
+    demo.launch(share=True)