Update app.py

app.py

@@ -11,7 +11,7 @@ retrieval_model_name = 'output/sentence-transformer-finetuned/'
 
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
-system_message = "You are an eloquent, dreamy and imaginative astronomy chatbot named Starfinder specialized in providing information on stargazing, astronomical events, and outer space."
+system_message = "You are an astronomy chatbot named Starfinder specialized in providing information on stargazing, astronomical events, and outer space."
 # Initial system message to set the behavior of the assistant
 messages = [{"role": "system", "content": system_message}]
 
@@ -73,9 +73,9 @@ def generate_response(user_query, relevant_segment):
         messages.append({"role": "user", "content": user_message})
         
         response = openai.ChatCompletion.create(
-            model="gpt-4o",
+            model="gpt-3.5-turbo",
             messages=messages,
-            max_tokens=500,
+            max_tokens=150,
             temperature=0.2,
             top_p=1,
             frequency_penalty=0,
@@ -100,16 +100,20 @@ def query_model(question):
     """
     if question == "":
         return "Welcome to Starfinder! Ask me anything about outer space, stargazing, and upcoming astronomical events."
+    
+    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"
+    
     relevant_segment = find_relevant_segment(question, segments)
     if not relevant_segment:
         return "Could not find specific information. Please refine your question."
     response = generate_response(question, relevant_segment)
-    return response
+    return response, None
 
 # Define the welcome message and specific topics the chatbot can provide information about
 welcome_message = """
 # ♟️ Welcome to Starfinder!
-
 ## Your AI-driven assistant for all astronomy-related queries. Created by Aarna, Aditi, and Anastasia of the 2024 Kode With Klossy SF Camp. 
 """
 
@@ -122,21 +126,9 @@ topics = """
 - Celestial events
 - Astronomy tips
 """
-STARS = gr.themes.Base().set(
-    background_fill_primary='#FFD700',  # Light yellow background
-    background_fill_primary_dark='#FFD700',  # Darker yellow background
-    background_fill_secondary='#3E92CC',  # Light blue background
-    background_fill_secondary_dark='#2A628F',  # Darker blue background
-    border_color_accent='#16324F',  # Accent border color (dark blue)
-    border_color_accent_dark='#16324F',  # Dark accent border color (dark blue)
-    border_color_accent_subdued='#18435A',  # Subdued accent border color (slightly lighter blue)
-    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)
-    button_primary_background_fill_dark='#2A628F'  # Dark primary button background color (medium blue)
-)
+
 # Setup the Gradio Blocks interface with custom layout components
-with gr.Blocks(theme=STARS) as demo:
+with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as demo:
     gr.Markdown(welcome_message)  # Display the formatted welcome message
     with gr.Row():
         with gr.Column():
@@ -145,9 +137,12 @@ with gr.Blocks(theme=STARS) as demo:
         with gr.Column():
             question = gr.Textbox(label="Your question", placeholder="What do you want to ask about?")
             answer = gr.Textbox(label="StarFinder Response", placeholder="StarFinder will respond here...", interactive=False, lines=10)
+            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)
+            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)
+
+