Update app.py

app.py

@@ -10,7 +10,7 @@ from langchain.llms import GooglePalm
 
 load_dotenv(find_dotenv())
 
-llm = GooglePalm(temperature=0.9, google_api_key=os.getenv("GOOGLE_API_KEY"))
+#llm = GooglePalm(temperature=0.9, google_api_key=os.getenv("GOOGLE_API_KEY"))
 
 # Iamge to Text
 def image_to_text(url):
@@ -23,21 +23,7 @@ def image_to_text(url):
     return text
 
 # llm
-def generate_story(scenario):
-    template = """
-    you are a very good story teller and a very rude person:
-    you can generate a short fairy tail based on a single narrative, the story should take 5 seconds to read.
-
-    CONTEXT: {scenario}
-    STORY:
-    """
-
-    prompt = PromptTemplate(template=template, input_variables=["scenario"])
-    story_llm = LLMChain(llm=llm, prompt=prompt, verbose=True)
-    story = story_llm.predict(scenario=scenario)
-    print(story)
-    return story
-
+#
 # text to speech
 
 def text_to_speech(message):
@@ -62,15 +48,11 @@ def main():
             file.write(bytes_data)
             st.image(uploaded_file, caption='Uploaded Image.', use_column_width=True)
         scenario = image_to_text(uploaded_file.name)
-        story = generate_story(scenario)
-        text_to_speech(story)
+        
 
         with st.expander("scenerio"):
             st.write(scenario)
-        with st.expander("story"):
-            st.write(story)
-
-        st.audio("audio.mp3")
+        
 
 if __name__ == '__main__':
     main()