Update app.py

app.py

@@ -49,7 +49,7 @@ def generate(
     message: str,
     chat_history: list[tuple[str, str]],
     max_new_tokens: int = 1024,
-    temperature: float = 0.1,  # Lower -> less random
+    temperature: float = 0.4,  # Lower -> less random
     top_p: float = 0.1,  # Lower -> less random, considering only the top 10% of tokens at each step
     top_k: int = 1,  # Least random, only the most likely next token is considered
     repetition_penalty: float = 1.0,  # No repetition penalty
@@ -87,6 +87,11 @@ def generate(
     for text in streamer:
         outputs.append(text)
         yield "".join(outputs)
+        
+    final_story = "".join(outputs) 
+    conversation_id = save_chat_history(chat_history + [(message, final_story)])  
+
+        yield f"Conversation ID: {conversation_id}"
 
 
 chat_interface = gr.ChatInterface(