Update app.py

app.py

@@ -91,7 +91,20 @@ def call_gemini(history: List[types.Content]):
     except Exception as e:
         print(f"GenAI Error: {e}")
         raise HTTPException(status_code=500, detail="AI Generation Failed")
+        
 
+def gen_title(user, model):
+    try:
+        response = client.models.generate_content(
+            model="gemma-3-1b-it",
+            contents= title_prompt(user, model)
+        )
+        return response.text
+    except Exception as e:
+        return user
+
+def title_prompt(user, model):
+    return f'''---\n**System Role:**\n\n> You are a specialized utility model. Your sole task is to generate a concise, descriptive title (2–6 words) for a chat conversation based on the provided user query and model response.\n\n**Constraints:**\n\n* **Output only the title.** Do not include labels, quotes, full-stop or introductory text.\n* **Focus on the core intent** of the conversation.\n* **Avoid generic terms** like "Chat about..." or "Request for..."\n* **Style:** Professional, catchy, and informative.\n\n---\n\n**User Query:** {user}\n\n**Model Response:** {model}'''
 
 def getSystemPrompt():
     text = ''
@@ -128,7 +141,7 @@ async def handleNewConv(new_conv: NewConv, token: str = Depends(verify_access_to
         raw_history.append(["user", new_conv.prompt])
         raw_history.append(["model", text])
         convs[conv_id] = raw_history
-        return {"title": new_conv.prompt, "text": text, "conv_id": conv_id}
+        return {"title": gen_title(new_conv.prompt,text), "text": text, "conv_id": conv_id}
 
 
 @app.post("/gen_resp")