Update app.py

app.py

@@ -9,6 +9,11 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 filename = "output_topic_details.txt"  # Path to the file storing chess-specific details
 retrieval_model_name = 'output/sentence-transformer-finetuned/'
 
+# Define paths to images
+path_to_sf_image = "output/sf.png"
+path_to_sacramento_image = "output/sacramento.png"
+path_to_la_image = "output/la.png"
+
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
 system_message = "You are an astronomy chatbot named Starfinder specialized in providing information on stargazing, astronomical events, and outer space."
@@ -99,15 +104,18 @@ def query_model(question):
     Process a question, find relevant information, and generate a response.
     """
     if question == "":
-        return "Welcome to Starfinder! Ask me anything about outer space, stargazing, and upcoming astronomical events."
+        return "Welcome to Starfinder! Ask me anything about outer space, stargazing, and upcoming astronomical events.", None
     
     if "san francisco" in question.lower():
-        # Return the image path and a message
-        return "Here is a picture of San Francisco!", "/path/to/your/san-francisco.jpg"
+        return "Here is a picture of San Francisco!", path_to_sf_image
+    if "sacramento" in question.lower():
+        return "Here is a picture of Sacramento!", path_to_sacramento_image
+    if "los angeles" in question.lower() or "la" in question.lower():
+        return "Here is a picture of Los Angeles!", path_to_la_image
     
     relevant_segment = find_relevant_segment(question, segments)
     if not relevant_segment:
-        return "Could not find specific information. Please refine your question."
+        return "Could not find specific information. Please refine your question.", None
     response = generate_response(question, relevant_segment)
     return response, None
 
@@ -126,7 +134,6 @@ topics = """
 - Celestial events
 - Astronomy tips
 """
-import gradio as gr
 
 STARS = gr.themes.Base().set(
     background_fill_primary='#2A628F',  # Light yellow background
@@ -139,7 +146,6 @@ STARS = gr.themes.Base().set(
     border_color_primary='#2A628F',  # Primary border color (medium blue)
     block_border_color='#2A628F',  # Block border color (medium blue)
     button_primary_background_fill='#2A628F',  # Primary button background color (medium blue)
-
 )
 
 # Setup the Gradio Blocks interface with custom layout components
@@ -155,9 +161,10 @@ with gr.Blocks(theme=STARS) as demo:
             image_output = gr.Image(label="Image Output")  # Add an Image component
             submit_button = gr.Button("Submit")
             submit_button.click(fn=query_model, inputs=question, outputs=[answer, image_output])  # Update outputs to include the image component
-    
 
 # Launch the Gradio app to allow user interaction
 demo.launch(share=True)
 
 
+
+