Update gradio1final.py

gradio1final.py

@@ -24,15 +24,26 @@ from langchain import OpenAI
 
 
 
-os.environ['GOOGLE_API_KEY'] = 'AIzaSyDLwJAr5iXL2Weaw1XphFNSeijytqOSbDg'
-os.environ['OPENAI_API_KEY'] = 'sk-proj-Kh4UIWkfDxSGppQpooxXT3BlbkFJATohXqhpkJE6MqliIkmU'
-# Set your API keys
+# os.environ['GOOGLE_API_KEY'] = 'AIzaSyDLwJAr5iXL2Weaw1XphFNSeijytqOSbDg'
+# os.environ['OPENAI_API_KEY'] = 'sk-proj-Kh4UIWkfDxSGppQpooxXT3BlbkFJATohXqhpkJE6MqliIkmU'
+# # Set your API keys
+
+# OPENAI_API_KEY = 'sk-proj-Kh4UIWkfDxSGppQpooxXT3BlbkFJATohXqhpkJE6MqliIkmU'
+# GOOGLE_API_KEY = 'AIzaSyDLwJAr5iXL2Weaw1XphFNSeijytqOSbDg'
+# openai.api_key = OPENAI_API_KEY
+# genai.configure(api_key=GOOGLE_API_KEY)
+
 
-OPENAI_API_KEY = 'sk-proj-Kh4UIWkfDxSGppQpooxXT3BlbkFJATohXqhpkJE6MqliIkmU'
-GOOGLE_API_KEY = 'AIzaSyDLwJAr5iXL2Weaw1XphFNSeijytqOSbDg'
-openai.api_key = OPENAI_API_KEY
-genai.configure(api_key=GOOGLE_API_KEY)
+google_api_key = os.environ.get('GOOGLE_API_KEY')
+openai_api_key = os.environ.get('OPENAI_API_KEY')
 
+# Ensure that the API keys are set
+if not google_api_key or not openai_api_key:
+    raise ValueError("API keys for Google and OpenAI must be set as environment variables.")
+
+# Set your API keys
+openai.api_key = openai_api_key
+genai.configure(api_key=google_api_key)
 
 
 
@@ -588,9 +599,8 @@ def process_gradio(query, model_type):
     except Exception as e:
         # Ensure the function still returns four values, even in case of an error
         return {}, str(e), "", []
-   
-image_path = r"C:\Users\shivanshu.t\Downloads\incedo-logo.png"
 
+image_path = r"C:\Users\shivanshu.t\Downloads\incedo-logo.png"
 
 with gr.Blocks() as demo:
     with gr.Row():
@@ -598,16 +608,8 @@ with gr.Blocks() as demo:
             # Add the image in the left corner
             # gr.Image(value=image_path, show_label=False, height=45, width=45)
             # gr.HTML(f'<img src="{image_link}" alt="Logo" height="50" width="50">')
-            # gr.Image(value=image_path, show_label=False, height=50, width=50)
+            gr.Image(value=image_path, show_label=False, height=50, width=50)
             # gr.HTML(f'<img src="{image_path}" alt="Logo" height="50" width="100" style="display: block; margin-left: auto; margin-right: auto;">')
-            # gr.HTML(f'<img src="https://mma.prnewswire.com/media/1807312/incedo_Logo.jpg" alt="Logo" height="50" width="80" style="display: block; margin-left: auto; margin-right: auto;">')
-            gr.HTML(
-                """
-                <div style="text-align: left; padding: 10px;">
-                    <img src="https://mma.prnewswire.com/media/1807312/incedo_Logo.jpg" alt="Logo" height="50" width="80">
-                </div>
-                """
-            )
     
     gr.Markdown(
         """