Update app.py

app.py

@@ -13,11 +13,7 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-openai_key = os.getenv("OPENAI_KEY")
-
-if openai_key == "":
-    sys.exit("Please Provide Your OpenAI API Key")
-
+openai_key = os.environ.get(OPENAI_KEY)
 
 def generate_image(text):
     try:
@@ -82,12 +78,8 @@ def make_prompt(entry):
     # 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):
-    # First, handle the specific case for [answer:]
-    # This replaces [answer:] with "Answer:" and keeps the content after it on the same line.
+
     text = re.sub(r'\[answer:\]\s*', 'Answer: ', text)
-    
-    # Now, remove all other content within brackets.
-    # This regex looks for square brackets and any content inside them, excluding those that start with "Answer: " already modified.
     text = re.sub(r'\[.*?\](?<!Answer: )', '', text)
     
     return text
@@ -160,7 +152,7 @@ def generate(
     #     yield f"Failed to save story: {str(e)}"
     
 
-# Gradio Interface Setup
+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,
     label=" ",
@@ -171,16 +163,15 @@ 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'
+    theme='shivi/calm_seafoam',autofocus=True, chatbot=chatbot
 )
 
 # Gradio Web Interface
 with gr.Blocks(css=custom_css,theme='shivi/calm_seafoam',fill_height=True) as demo:
-    output_image = gr.Image(label="Image Output")
-    chat_interface.render()
+        chat_interface.render()
     # gr.Markdown(LICENSE)
 
-
+        
 # Main Execution
 if __name__ == "__main__":
     demo.queue(max_size=20)