Update app.py

app.py

@@ -202,7 +202,7 @@ def generate(PROMPT,model):
 
 ########################
 API_ENDPOINT = "https://api.imgbb.com/1/upload"
-API_KEY = os.environ('IMAGE_API_KEY')
+API_KEY = os.environ['IMAGE_API_KEY']
 
 
 def imgLink(image):
@@ -237,7 +237,7 @@ def google_search(image):
   "engine": "google_lens",
   "url": image_url,
   "hl": "en",
-  "api_key": os.environ('GOOGLE_SEARCH_API_KEY')
+  "api_key": os.environ['GOOGLE_SEARCH_API_KEY']
   }
   search = GoogleSearch(params)
   result = search.get_dict()
@@ -688,14 +688,16 @@ home</p>""")
   with gr.Accordion("Generate Knowledge Graph",open=False):
     gr.Markdown("**Genrate Knowledge Graph**")
     gr.Markdown("**Refer to the example below**")  
-    gr.HTML(value="<img src='https://i.ibb.co/j3j5Y3Q/4f4f9ed9-b52d-4ed9-a5e8-5864bffd4718.jpg' alt='Generate knowlwdge graph' width='1200' height='300'/>")   
     with gr.Row():
       with gr.Column():
         prompt_to_generate_graph = gr.Text()
       with gr.Column():
         generated_graph_pic = gr.Image()
+    gr.HTML(value="<img src='https://i.ibb.co/j3j5Y3Q/4f4f9ed9-b52d-4ed9-a5e8-5864bffd4718.jpg' alt='Generate knowlwdge graph' width='1200' height='300' style='border: 2px solid #000;'/>")   
+    
 
     generate_knowledge_graph = gr.Button("Generate Knowledge Graph")
     generate_knowledge_graph.click(knowledge_graph, inputs=prompt_to_generate_graph, outputs=generated_graph_pic)
+      
    
 demo.launch(debug=True)