Update app.py

app.py

@@ -7,10 +7,10 @@ from collections import defaultdict
 import os
 import re
 
-# Load environment variables (includes OPENAI_API_KEY)
+# Load environment variables (like OPENAI_API_KEY)
 load_dotenv(override=True)
 
-# Track daily question count per user IP
+# Track questions per IP
 user_question_counter = defaultdict(lambda: {"date": None, "count": 0})
 
 
@@ -19,7 +19,7 @@ class Me:
         self.openai = OpenAI()
         self.name = "Narendra"
 
-        # Load LinkedIn text
+        # Load LinkedIn profile
         reader = PdfReader("me/linkedin.pdf")
         self.linkedin = ""
         for page in reader.pages:
@@ -27,26 +27,26 @@ class Me:
             if text:
                 self.linkedin += text
 
-        # Load professional summary
+        # Load summary
         with open("me/summary.txt", "r", encoding="utf-8") as f:
             self.summary = f.read()
 
     def system_prompt(self):
         return (
             f"You are acting as {self.name}, an experienced Python technical interviewer. "
-            f"Only answer questions related to Python programming — skip all non-Python topics. "
-            f"Do not exceed 100 tokens in your response. "
-            f"\n\n## About {self.name}:\n{self.summary}\n\n"
+            f"Only answer questions related to Python programming. If the question is unrelated to Python, say so. "
+            f"All answers must be under 100 tokens.\n\n"
+            f"## About {self.name}:\n{self.summary}\n\n"
             f"## LinkedIn Profile:\n{self.linkedin}"
         )
 
     def is_python_related(self, text):
-        """Smarter keyword-based filter for Python-related questions"""
         python_keywords = [
             "python", "list", "dictionary", "dict", "tuple", "set", "loop",
             "for", "while", "comprehension", "function", "class", "exception",
             "PEP", "decorator", "lambda", "flask", "django", "pandas", "numpy",
-            "jupyter", "interpreter", "import", "package", "virtualenv", "pytest"
+            "jupyter", "interpreter", "import", "package", "virtualenv", "pytest",
+            "recursion", "palindrome", "factorial", "generator", "iterator"
         ]
         text_lower = text.lower()
         return any(kw in text_lower for kw in python_keywords)
@@ -56,7 +56,6 @@ class Me:
         today = datetime.date.today()
         record = user_question_counter[ip]
 
-        # Reset daily count if new day
         if record["date"] != today:
             record["date"] = today
             record["count"] = 0
@@ -67,20 +66,23 @@ class Me:
         if not self.is_python_related(message):
             return "⚠️ I can only answer questions related to Python programming. Please ask something Python-specific."
 
-        # Build messages for OpenAI
         messages = [{"role": "system", "content": self.system_prompt()}] + history + [{"role": "user", "content": message}]
-        
         response = self.openai.chat.completions.create(
             model="gpt-4o-mini",
             messages=messages,
-            max_tokens=100
+            max_tokens=50
         )
 
         record["count"] += 1
-        return f"👋 Narendra is your Python interviewer. Let's begin!\n\n{response.choices[0].message.content}"
+        return response.choices[0].message.content
 
 
 if __name__ == "__main__":
     me = Me()
-    # Pass `request` to get user IP
-    gr.ChatInterface(me.chat, type="messages", additional_inputs=[], concurrency_limit=None).launch(share=True)
+    gr.ChatInterface(
+        fn=me.chat,
+        type="messages",
+        title="👋 Narendra is your Python Interviewer. Ask your Python questions (max 3/day).",
+        concurrency_limit=None,
+        theme="default"
+    ).launch(share=True, show_api=False)