Update chatbot_page.py

chatbot_page.py

@@ -36,14 +36,28 @@ def chat_with_user(user_message, history):
 
 # Function to end chat and extract keywords
 def extract_keywords_from_conversation(history):
+    from openai import OpenAI
+    client = OpenAI()
     # Combine all user and assistant messages into a single string
     conversation = "\n".join([f"User: {msg[0]}\nAssistant: {msg[1]}" for msg in history if msg[1]])
-    prompt = (
-        "Given the following conversation between a user and an assistant about finding an ideal open-source repo, "
-        "extract about 5 keywords that would be most useful for searching Hugging Face repos to find the most relevant results for the user. "
-        "Return only the keywords as a comma-separated list.\n\nConversation:\n" + conversation
+    system_prompt = (
+        "You are an expert at helping users find open-source repos on Hugging Face. "
+        "Given a conversation, extract about 5 keywords that would be most useful for searching Hugging Face repos to find the most relevant results for the user. "
+        "Return only the keywords as a comma-separated list."
+    )
+    user_prompt = (
+        "Conversation:\n" + conversation + "\n\nExtract about 5 keywords for Hugging Face repo search."
+    )
+    response = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_prompt}
+        ],
+        max_tokens=64,
+        temperature=0.3
     )
-    keywords = analyze_code(prompt)
+    keywords = response.choices[0].message.content.strip()
     return keywords
 
 with gr.Blocks() as chatbot_demo: