Update app.py

app.py

@@ -3,6 +3,8 @@ import os
 import gradio as gr
 import google.generativeai as genai  # Importing the Gemini API module
 
+# Initialize the Gemini model
+Model = genai.GenerativeModel('gemini-1.5-flash')  # Initialize the generative model
 
 def chat_short_story(length, genre, theme, tone, writing_style):
     """
@@ -37,11 +39,9 @@ def chat_short_story(length, genre, theme, tone, writing_style):
 
     # Generate a response using the Gemini model
     try:
-        response = genai.generate_text(prompt=prompt, model="gemini-1.5-flash")
-        if "candidates" in response and len(response["candidates"]) > 0:
-            return response["candidates"][0]["output"]  # Extract the story content
-        else:
-            return "Error: No story generated. Please check your input or try again."
+        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
     except Exception as e:
         return f"Error: Failed to generate story. Details: {e}"