Update app.py

app.py

@@ -1,9 +1,6 @@
 import os
 import requests
 import gradio as gr
-import base64
-import io
-from PIL import Image
 
 #
 # 1) Retrieve your Hugging Face token (with read access) for the private space
@@ -54,24 +51,21 @@ def handle_feedback(user_query, answer, feedback_option):
     return call_private_space(1, [user_query, answer, feedback_option])
 
 #
-
-#
-# 6) Build the UI *layout* to mirror your private space.
+# 5) Build the UI layout to mirror your private space,
+#    now loading dal_logo.png locally from the *public* repo.
 #
-demo = gr.Blocks()
-
-with demo:
-    # A row with the logo on the left, and a title on the right
+with gr.Blocks() as demo:
+    # Top row: logo on the left, title on the right
     with gr.Row():
         with gr.Column(scale=1, min_width=100):
-            if DAL_LOGO_IMG:
-                gr.Image(
-                    value=DAL_LOGO_IMG,
-                    show_label=False,
-                    interactive=False,
-                    width=80,
-                    height=80
-                )
+            # Load the local dal_logo.png (make sure it's in the same folder!)
+            gr.Image(
+                value="dal_logo.png",
+                show_label=False,
+                interactive=False,
+                width=80,
+                height=80
+            )
         with gr.Column(scale=6):
             gr.Markdown("## Dallas RAG + Graph Demo")
             gr.Markdown(
@@ -79,8 +73,8 @@ with demo:
                 "You can **toggle** whether the subgraph is summarized via OpenAI "
                 "or used 'as is'. Only the last 5 queries are kept in the conversation."
             )
-    
-    # User input row
+
+    # User query, checkbox, and 'Get Answer'
     user_query = gr.Textbox(
         label="Your Query",
         lines=1,
@@ -91,9 +85,11 @@ with demo:
         label="Use Graph Summaries?"
     )
     get_answer_btn = gr.Button("Get Answer")
+
+    # Output area
     answer_output = gr.Markdown(label="AI Answer")
 
-    # Feedback row
+    # Feedback section
     feedback_radio = gr.Radio(
         choices=["👍", "⚖️", "👎"],
         value="⚖️",
@@ -102,7 +98,7 @@ with demo:
     feedback_btn = gr.Button("Submit Feedback")
     feedback_result = gr.Markdown()
 
-    # Wire up the .click() events to the private space calls
+    # Wire the buttons to the private space endpoints
     get_answer_btn.click(
         fn=run_query,
         inputs=[user_query, use_summary_chk],
@@ -114,7 +110,7 @@ with demo:
         outputs=feedback_result
     )
 
-    # Footer, same as private space
+    # Footer
     gr.HTML(
         """
         <div style="text-align: center; margin-top: 20px;">
@@ -127,8 +123,6 @@ with demo:
         """
     )
 
-#
-# 7) Launch
-#
+# Launch the public interface
 if __name__ == "__main__":
     demo.launch()