Update app.py

app.py

@@ -60,20 +60,9 @@ if torch.cuda.is_available():
     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)
-
-# Utility function for prompts
 def make_prompt(entry):
     return  f"### Human: When asked to explain use a story.Don't repeat the assesments, limit to 500 words.However keep context in mind if edits to the content is required. {entry} ### Assistant:"
-    # f"TELL A STORY, RELATE TO COMPUTER SCIENCE, INCLUDE ASSESMENTS. MAKE IT REALISTIC AND AROUND 800 WORDS, END THE STORY WITH "THE END.": {entry}"
 
 def process_text(text):
 
@@ -142,15 +131,12 @@ def generate(
         yield output
 
     final_story = "".join(outputs)
-    generate_image(final_story)
-    # try:
-    #     saved_story = Story(message=message, content=final_story).save()
-    #     yield f"{final_story}\n\n Story saved with ID: {saved_story.story_id}"
-    # except Exception as e:
-    #     yield f"Failed to save story: {str(e)}"
-    
+    image_url = generate_image(final_story)
+    yield image_url
+
+
+
 
-chatbot=gr.Chatbot([(None, ("https://img.freepik.com/free-photo/painting-mountain-lake-with-mountain-background_188544-9126.jpg",))])
 chat_interface = gr.ChatInterface(
     fn=generate,
     fill_height=True,
@@ -160,7 +146,7 @@ chat_interface = gr.ChatInterface(
         ["Could you please provide an explanation about the concept of recursion?"],
         ["Could you explain what a URL is?"]
     ],
-    theme='shivi/calm_seafoam',autofocus=True, chatbot=chatbot
+    theme='shivi/calm_seafoam',autofocus=True
 )
 
 # Gradio Web Interface