Update app.py

app.py

@@ -1,26 +1,29 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load your fine-tuned Corporate Leadership model from Hugging Face
+# Load the fine-tuned Corporate Leadership model from Hugging Face
 generator = pipeline("text-generation", model="shibly100/gpt2_corporate_leadership")
 
-# Define chatbot function
-def chat_with_ai(prompt):
-    response = generator(prompt, max_length=200, do_sample=True)
-    return response[0]["generated_text"]
+# Define the chat function that remembers conversation history
+def chat_with_ai(history, message):
+    full_prompt = " ".join([turn[0] + " " + turn[1] for turn in history] + [message])
+    response = generator(full_prompt, max_length=300, do_sample=True, temperature=0.7)
+    reply = response[0]["generated_text"]
+    
+    # Ensure only new AI-generated response is returned
+    reply = reply[len(full_prompt):].strip()
+    history.append((message, reply))
+    return history, ""
 
-# Create the Gradio Chatbot Interface
-demo = gr.Interface(
-    fn=chat_with_ai,
-    inputs=gr.Textbox(lines=3, placeholder="Ask anything about corporate leadership..."),
-    outputs="text",
-    title="SPARKY Corporate Leadership AI",
-    description="Ask anything about corporate leadership and get AI-powered insights!",
-    allow_flagging="never",
+# Create Chatbot Interface
+chatbot = gr.ChatInterface(
+    chat_with_ai,
+    title="🗣️ SPARKY Corporate Leadership AI",
+    description="🤖 Ask anything about corporate leadership and get AI-powered responses! Your conversation history is remembered.",
+    theme="soft",  # This gives it a sleek, modern look like ChatGPT
 )
 
 # Launch the chatbot
-demo.launch()
-
+chatbot.launch()