Update app.py

app.py

@@ -43,15 +43,15 @@ model = PeftModel.from_pretrained(base_model, "ranamhamoud/storytell")
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 tokenizer.pad_token = tokenizer.eos_token
 
-# MongoDB Connection
-PASSWORD = os.environ.get("MONGO_PASS")
-connect(host=f"mongodb+srv://ranamhammoud11:{PASSWORD}@stories.zf5v52a.mongodb.net/")
-
-# MongoDB Document
-class Story(Document):
-    message = StringField()
-    content = StringField()
-    story_id = SequenceField(primary_key=True)
+# # MongoDB Connection
+# PASSWORD = os.environ.get("MONGO_PASS")
+# connect(host=f"mongodb+srv://ranamhammoud11:{PASSWORD}@stories.zf5v52a.mongodb.net/")
+
+# # MongoDB Document
+# class Story(Document):
+#     message = StringField()
+#     content = StringField()
+#     story_id = SequenceField(primary_key=True)
    
 
 # Utility function for prompts
@@ -135,8 +135,8 @@ def generate(
     final_story_trimmed = remove_last_sentence(final_story)
 
     try:
-        saved_story = Story(message=message, content=final_story_trimmed).save()
-        yield f"{final_story_trimmed}\n\n Story saved with ID: {saved_story.story_id}"
+        # saved_story = Story(message=message, content=final_story_trimmed).save()
+        yield f"{final_story_trimmed}"
     except Exception as e:
         yield f"Failed to save story: {str(e)}"