Update app.py

app.py

@@ -20,7 +20,7 @@ def llava(inputs, history):
     return processed
 
 def respond(message, history):
-    """Generate a response based on input."""
+    """Generate a response for input."""
     if "files" in message and message["files"]:
         inputs = llava(message, history)
         streamer = TextIteratorStreamer(skip_prompt=True, skip_special_tokens=True)
@@ -44,31 +44,40 @@ def generate_image(prompt):
     client = InferenceClient("KingNish/Image-Gen-Pro")
     return client.predict("Image Generation", None, prompt, api_name="/image_gen_pro")
 
+# State management to control visibility
+def show_page(page, state):
+    """Updates the state to show the selected page."""
+    return {"chat_visible": page == "chat", "image_visible": page == "image"}
+
 # Gradio app setup
 with gr.Blocks(title="AI Chat & Tools") as demo:
+    state = gr.State({"chat_visible": True, "image_visible": False})
+
     with gr.Row():
         with gr.Column(scale=1, min_width=200):
             gr.Markdown("## Navigation")
-            gr.Button("Chat Interface").click(None, [], [], _js="() => window.location.hash='#chat'")
-            gr.Button("Image Generation").click(None, [], [], _js="() => window.location.hash='#image'")
-        
+            chat_button = gr.Button("Chat Interface")
+            image_button = gr.Button("Image Generation")
+
         with gr.Column(scale=3):
-            with gr.Page("chat"):
+            with gr.Row(visible=lambda state: state["chat_visible"], interactive=True):
                 gr.Markdown("## Chat with AI Assistant")
                 chatbot = gr.Chatbot(label="Chat", show_label=False)
-                with gr.Row():
-                    text_input = gr.Textbox(placeholder="Enter your message...", lines=2, show_label=False)
-                    file_input = gr.File(label="Upload an image", file_types=["image/*"])
-                
+                text_input = gr.Textbox(placeholder="Enter your message...", lines=2, show_label=False)
+                file_input = gr.File(label="Upload an image", file_types=["image/*"])
                 text_input.submit(respond, [text_input, chatbot], [chatbot])
                 file_input.change(respond, [file_input, chatbot], [chatbot])
-            
-            with gr.Page("image"):
+
+            with gr.Row(visible=lambda state: state["image_visible"], interactive=True):
                 gr.Markdown("## Image Generator")
                 image_prompt = gr.Textbox(placeholder="Describe the image to generate", show_label=False)
                 image_output = gr.Image(label="Generated Image")
-                
                 image_prompt.submit(generate_image, [image_prompt], [image_output])
 
+    # Button actions to switch between pages
+    chat_button.click(lambda: show_page("chat", state.value), None, state)
+    image_button.click(lambda: show_page("image", state.value), None, state)
+
 # Launch the app
 demo.launch()
+