Changed made in interface

app.py

@@ -4,6 +4,7 @@ from groq import Groq
 import gradio as gr
 
 load_dotenv()
+
 client = Groq()
 session_memory_dict = {}
 
@@ -20,110 +21,41 @@ class ChatInput(BaseModel):
     session_id: str
     input: str
 
-def chat_logic(session_id: str, user_input: str):
+def chat_logic(session_id: str, user_input: str) -> str:
     if session_id not in session_memory_dict:
         session_memory_dict[session_id] = []
 
     if not any(m["role"] == "system" for m in session_memory_dict[session_id]):
         session_memory_dict[session_id].insert(0, system_prompt)
 
-    session_memory_dict[session_id].append({"role": "user", "content": user_input})
-    res = client.chat.completions.create(
+    session_memory_dict[session_id].append({
+        "role": "user",
+        "content": user_input
+    })
+
+    completion = client.chat.completions.create(
         model="llama3-8b-8192",
         messages=session_memory_dict[session_id]
     )
-    ai_response = res.choices[0].message.content
-    session_memory_dict[session_id].append({"role": "assistant", "content": ai_response})
-    print(f"[{session_id}] AI:", ai_response)
-    return ai_response
-
-def chat_interface(user_input, chat_history, session_id):
-    ai_reply = chat_logic(session_id, user_input)
-    chat_history.append((user_input, ai_reply))
-    return chat_history, ""
-
-def clear_session(session_id):
-    session_memory_dict.pop(session_id, None)
-    return [], "gradio_default"
-
-with gr.Blocks(css="""
-body {
-  background-color: #1e1e1e;
-  color: #ddd;
-  font-family: 'Segoe UI', sans-serif;
-}
-.header {
-  text-align: center;
-  font-size: 1.8rem;
-  padding: 1rem;
-  background-color: #2c2c2c;
-  color: #ffffff;
-  border-bottom: 1px solid #444;
-}
-.chatbot {
-  flex: 1;
-  display: flex;
-  flex-direction: column;
-  height: 65vh;
-  overflow-y: auto;
-  padding: 1rem;
-}
-.chatbot .message {
-  margin: 6px 0;
-  white-space: pre-wrap;
-  word-wrap: break-word;
-  line-height: 1.6;
-  font-size: 1rem;
-}
-.chatbot .message.user {
-  align-self: flex-end;
-  color: #90caf9;
-}
-.chatbot .message.bot {
-  align-self: flex-start;
-  color: #e0e0e0;
-}
-.input-section {
-  display: flex;
-  gap: 10px;
-  align-items: center;
-  margin-top: auto;
-}
-.send-button {
-  background-color: #2979FF;
-  color: white;
-  padding: 10px 16px;
-  border-radius: 8px;
-  border: none;
-  cursor: pointer;
-}
-.send-button:hover {
-  background-color: #5393FF;
-}
-.session-controls {
-  display: flex;
-  justify-content: space-between;
-  gap: 10px;
-  margin-top: 1rem;
-}
-""") as demo:
-
-    gr.Markdown("## 🤖 Smart Assistant", elem_classes="header")
-
-    state = gr.State([])
 
-    chatbot = gr.Chatbot(elem_classes="chatbot")
+    ai_response = completion.choices[0].message.content
 
-    with gr.Row(elem_classes="input-section"):
-        user_input = gr.Textbox(placeholder="Type your message...", lines=1, label="")
-        send_btn = gr.Button("Send", elem_classes="send-button")
+    session_memory_dict[session_id].append({
+        "role": "assistant",
+        "content": ai_response
+    })
 
-    with gr.Row(elem_classes="session-controls"):
-        session_id = gr.Textbox(label="Session ID", value="gradio_default", interactive=True)
-        clear_btn = gr.Button("Clear Chat")
-
-    send_btn.click(fn=chat_interface, inputs=[user_input, state, session_id], outputs=[chatbot, user_input])
-    user_input.submit(fn=chat_interface, inputs=[user_input, state, session_id], outputs=[chatbot, user_input])
-    clear_btn.click(fn=clear_session, inputs=[session_id], outputs=[chatbot, session_id])
+    print(f"[Session: {session_id}] AI Response: {ai_response}")
+    return ai_response
 
-    demo.launch()
+# Gradio Interface
+def gradio_chat(user_input, session_id="gradio_default"):
+    return chat_logic(session_id=session_id, user_input=user_input)
+
+gr.Interface(
+    fn=gradio_chat,
+    inputs=[gr.Textbox(label="Your message"), gr.Textbox(label="Session ID", value="gradio_default")],
+    outputs=gr.Textbox(label="Response"),
+    title="Chatbot with memory",
+    description="Chat with Groq's LLaMA3 model. Handles sessions separately using IDs."
+).launch()