Upload Langchain.py

pages/Langchain.py

@@ -0,0 +1,99 @@
+import json
+import datetime
+import sqlite3
+import asyncio
+import requests
+import conteneiro
+import websockets
+import streamlit as st
+from agentLangchain import langchainAgent
+from Classifier import IntentClassifier
+
+servers = []
+clients = []
+inputs = []
+outputs = []
+messagess = []
+intentios = []
+used_ports = []
+server_ports = []
+client_ports = []
+
+db = sqlite3.connect('chat-hub.db')
+cursor = db.cursor()
+cursor.execute('CREATE TABLE IF NOT EXISTS messages (id INTEGER PRIMARY KEY AUTOINCREMENT, sender TEXT, message TEXT, timestamp TEXT)')    
+db.commit()
+
+async def main():
+
+    userInput = st.chat_input("Ask Agent")         
+
+    c1, c2 = st.columns(2)
+    fireAPI = st.text_input("Fireworks API")
+
+    with c1:
+        stat1 = st.empty()
+        state1 = stat1.status(label="Langchain", state="complete", expanded=False)
+        state1.write(conteneiro.servers)
+        websocketPort = st.number_input("Websocket servers", min_value=1000, max_value=9999, value=1000)   
+        startServer = st.button("Start server")
+        stopServer = st.button("Stop server")            
+
+    with c2:
+        stat2 = st.empty()
+        state2 = stat2.status(label="Langchain", state="complete", expanded=False)
+        state2.write(conteneiro.clients)
+        clientPort = st.number_input("Websocket clients", min_value=1000, max_value=9999, value=1000)   
+        start_Client = st.button("Start client")
+        stopClient = st.button("Stop client")       
+
+    with st.sidebar:
+        cont = st.empty()        
+        status = cont.status(label="Langchain", state="complete", expanded=False)
+        
+
+    if userInput:
+        user_input = st.chat_message("human")
+        user_input.markdown(userInput)
+        messagess.append(user_input)
+        agent = langchainAgent(fireAPI)
+        response = await agent.askQuestion(userInput)
+        outputMsg = st.chat_message("ai") 
+        outputMsg.markdown(response)
+        await agent.handleInput(response)
+        
+    if start_Client:
+        voiceCli = f"Langchain client port: {clientPort}"
+        stat2.empty()            
+        state2 = stat2.status(label=voiceCli, state="running", expanded=True)
+        state2.write(conteneiro.clients)
+        cont.empty()            
+        status = cont.status(label=voiceCli, state="running", expanded=True)
+        status.write(conteneiro.servers) 
+        try:
+            client = langchainAgent(fireAPI)
+            await client.startClient(clientPort)
+            print(f"Connecting client on port {clientPort}...")
+            await asyncio.Future()
+
+        except Exception as e:
+            print(f"Error: {e}")   
+
+    if startServer:
+        vooiceSrv = f"Langchain server pport: {websocketPort}"
+        stat1.empty()
+        state1 = stat1.status(label=vooiceSrv, state="running", expanded=False)
+        state1.write(conteneiro.clients)
+        cont.empty()            
+        status = cont.status(label=vooiceSrv, state="running", expanded=False)
+        status.write(conteneiro.clients)          
+        try:          
+            server = langchainAgent(fireAPI)
+            await server.start_server(websocketPort)                
+            print(f"Starting WebSocket server on port {websocketPort}...")
+            await asyncio.Future()
+
+        except Exception as e:
+            print(f"Error: {e}")  
+
+asyncio.run(main())