Update app.py

app.py

@@ -1,10 +1,10 @@
 import random
 import os
 import gradio as gr
-import google.generativeai as genai  # Importing the Gemini API module
+import google.generativeai as genai  
 
 # Initialize the Gemini model
-Model = genai.GenerativeModel('gemini-1.5-flash')  # Initialize the generative model
+Model = genai.GenerativeModel('gemini-1.5-flash')  
 
 def chat_short_story(length, genre, theme, tone, writing_style):
     """
@@ -39,9 +39,9 @@ def chat_short_story(length, genre, theme, tone, writing_style):
 
     # Generate a response using the Gemini model
     try:
-        chat = Model.start_chat()  # Start a chat session
-        response = chat.send_message(prompt)  # Send the prompt to the model
-        return response.text  # Extract the content of the response
+        chat = Model.start_chat()  
+        response = chat.send_message(prompt)  
+        return response.text  
     except Exception as e:
         return f"Error: Failed to generate story. Details: {e}"
 
@@ -80,7 +80,7 @@ iface = gr.Interface(
         gr.Dropdown(label="Story Tone", choices=Tones)
     ],
     outputs=gr.Textbox(label="Generated Story"),
-    title="Patrick's Story Generator",
+    title="Welcome to Patrick's Story Generator",
     description="Generate creative short stories tailored to your preferences."
 )