Update app.py

app.py

@@ -95,19 +95,21 @@ body, input, button, textarea, label {
 }
 """
 
-# Gradio Function
+def process_text(text):
+    text = re.sub(r'\[assessment;[^\]]*\]', '', text, flags=re.DOTALL)
+    text = re.sub(r'\[.*?\]', '', text, flags=re.DOTALL)
+    return text
+    
 @spaces.GPU
 def generate(
     message: str,
     chat_history: list[tuple[str, str]],
     max_new_tokens: int = DEFAULT_MAX_NEW_TOKENS,
-    temperature: float = 0.6,
+    temperature: float = 0.8,
     top_p: float = 0.7,
-    top_k: int = 20,
+    top_k: int = 30,
     repetition_penalty: float = 1.0,
 ) -> Iterator[str]:
-    logging.debug("Starting chat generation with message: %s", message)
-
     conversation = []
     for user, assistant in chat_history:
         conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
@@ -141,10 +143,17 @@ def generate(
         yield output
 
     final_story = "".join(outputs)
-    image_url = generate_image(final_story)
+    final_story_trimmed = remove_last_sentence(final_story)
+    
+    image_url = generate_image(final_story_trimmed)
     return f"{final_story}\n\n![Generated Image]({image_url})"
 
 
+def remove_last_sentence(text):
+    # Assuming sentences end with a period followed by space or end of string
+    sentences = re.split(r'(?<=\.)\s', text)
+    return ' '.join(sentences[:-1]) if sentences else text
+
 chat_interface = gr.ChatInterface(
     fn=generate,
     fill_height=True,