Update app.py

app.py

@@ -6,7 +6,7 @@ import torch
 # Load Fine-Tuned GPT-2 Model
 # =========================
 def load_model():
-    model_path = "zaid002/finetunedmodel"   # Hugging Face repo
+    model_path = "zaid002/finetunedmodel"  # Hugging Face repo
     tokenizer = GPT2Tokenizer.from_pretrained(model_path)
     model = GPT2LMHeadModel.from_pretrained(model_path)
 
@@ -22,48 +22,52 @@ def load_model():
 tokenizer, chatbot = load_model()
 
 # =========================
-# ChatGPT-style Prompt
+# ChatGPT System Prompt
 # =========================
 SYSTEM_PROMPT = """
-You are an intelligent, multilingual AI assistant like ChatGPT.
-You can answer questions related to:
-- GUVI courses
+You are a powerful AI assistant like ChatGPT.
+
+You can answer:
+- GUVI course-related questions
 - Programming (Python, Java, SQL, ML, AI, Data Science)
 - Interview preparation
+- Career guidance
 - General knowledge
-- Education and career guidance
 
-You can reply in ANY language the user asks.
-Give clear, helpful, and human-like answers.
+Rules:
+- Respond in the SAME language as the user
+- Be clear, helpful, and human-like
+- Give structured answers when needed
+- Avoid repetition
 """
 
 # =========================
-# Chat Function
+# Chat Function (ChatGPT Style)
 # =========================
-def chatgpt_style_chat(user_input, history):
-    conversation = SYSTEM_PROMPT + "\n"
+def chatgpt_multilingual(user_input, history):
+    prompt = SYSTEM_PROMPT + "\n\n"
 
-    # Add previous chat history
+    # Add conversation memory
     for user, bot in history:
-        conversation += f"User: {user}\nAssistant: {bot}\n"
+        prompt += f"User: {user}\nAssistant: {bot}\n"
 
-    # Add new user question
-    conversation += f"User: {user_input}\nAssistant:"
+    prompt += f"User: {user_input}\nAssistant:"
 
     output = chatbot(
-        conversation,
-        max_length=300,
+        prompt,
+        max_length=400,
         do_sample=True,
         temperature=0.7,
         top_p=0.9,
         top_k=50,
+        repetition_penalty=1.2,
         pad_token_id=tokenizer.eos_token_id
     )
 
     response = output[0]["generated_text"].split("Assistant:")[-1].strip()
 
-    # Prevent overly long or repeated answers
-    response = response[:500]
+    # Safety trim
+    response = response[:600]
 
     history.append((user_input, response))
     return history, history
@@ -72,29 +76,28 @@ def chatgpt_style_chat(user_input, history):
 # Gradio UI
 # =========================
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.HTML(
-        """
-        <div style="text-align:center;">
-            <h1>🤖 GUVI Multilingual ChatGPT Bot</h1>
-            <p>Ask anything about GUVI, programming, careers, or general topics — in ANY language.</p>
-        </div>
-        """
-    )
+    gr.HTML("""
+    <div style="text-align:center;">
+        <h1>🤖 GUVI Multilingual ChatGPT Bot</h1>
+        <p>Ask anything about GUVI, programming, careers, or general topics — in ANY language.</p>
+    </div>
+    """)
 
     chatbot_ui = gr.Chatbot(height=450)
+
     msg = gr.Textbox(
-        placeholder="💬 Ask anything (English / हिंदी / தமிழ் / తెలుగు / Français / Español / etc.)",
+        placeholder="💬 Ask in English / हिंदी / தமிழ் / తెలుగు / Français / Español / etc.",
         label="Your Message"
     )
 
     state = gr.State([])
     clear = gr.Button("🗑️ Clear Chat")
 
-    msg.submit(chatgpt_style_chat, [msg, state], [chatbot_ui, state])
+    msg.submit(chatgpt_multilingual, [msg, state], [chatbot_ui, state])
     clear.click(lambda: ([], []), None, [chatbot_ui, state], queue=False)
 
 # =========================
-# Launch
+# Launch App
 # =========================
 if __name__ == "__main__":
     demo.launch()