Changed made in interface

app.py

@@ -21,6 +21,7 @@ class ChatInput(BaseModel):
     session_id: str
     input: str
 
+# Chat logic with memory handling
 def chat_logic(session_id: str, user_input: str) -> str:
     if session_id not in session_memory_dict:
         session_memory_dict[session_id] = []
@@ -28,10 +29,7 @@ def chat_logic(session_id: str, user_input: str) -> str:
     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
-    })
+    session_memory_dict[session_id].append({"role": "user", "content": user_input})
 
     completion = client.chat.completions.create(
         model="llama3-8b-8192",
@@ -40,22 +38,22 @@ def chat_logic(session_id: str, user_input: str) -> str:
 
     ai_response = completion.choices[0].message.content
 
-    session_memory_dict[session_id].append({
-        "role": "assistant",
-        "content": ai_response
-    })
-
+    session_memory_dict[session_id].append({"role": "assistant", "content": ai_response})
     print(f"[Session: {session_id}] AI Response: {ai_response}")
     return ai_response
 
-# 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()
+# Gradio ChatInterface function wrapper
+def chat_with_memory(message, history, session_id="gradio_default"):
+    ai_reply = chat_logic(session_id=session_id, user_input=message)
+    return ai_reply
+
+# UI Components
+chat_interface = gr.ChatInterface(
+    fn=chat_with_memory,
+    additional_inputs=[gr.Textbox(label="Session ID", value="gradio_default")],
+    title="Friendly Chatbot with Memory",
+    description="Chat with Groq's LLaMA3 model. Each session is tracked using a session ID.",
+    theme="soft"
+)
+
+chat_interface.launch()