Update app.py

Changed the interface : chat generation now chat completions .

Model querying is now done by the backend and users now select model from dropdown window

app.py

@@ -37,12 +37,11 @@ def generate_media(prompt, model_name, media_type):
 # Create a Gradio interface
 with gr.Blocks() as demo:
     with gr.Tab("Chat"):
-        with gr.Row():
-            language_model_input = gr.Textbox(label="Language Model")
-            query_button = gr.Button("Query HuggingFace Hub")
+        language_model_input = gr.Dropdown(label="Language Model", choices=[model.modelId for model in api.list_models()])
         chat_input = gr.Textbox(label="Chat Input")
-        chat_output = gr.Textbox(label="Chat Output")
+        chat_output = gr.Textbox(label="Chat Output", interactive=False)
         generate_button = gr.Button("Generate Text")
+        chat_history = gr.Chatbot(label="Chat History")
     
     with gr.Tab("Image Generation"):
         image_input = gr.Textbox(label="Image Prompt")
@@ -57,11 +56,8 @@ with gr.Blocks() as demo:
         generate_media_button = gr.Button("Generate Media")
         media_output = gr.Video(label="Generated Media") if media_type_input == "video" else gr.Audio(label="Generated Media")
     
-    # Query Hugging Face Hub for language models
-    query_button.click(fn=lambda x: [model.modelId for model in api.list_models(filter=x)], inputs=language_model_input, outputs=language_model_input)
-    
     # Generate text with a language model
-    generate_button.click(fn=generate_text, inputs=[language_model_input, chat_input], outputs=chat_output)
+    generate_button.click(fn=generate_text, inputs=[language_model_input, chat_input], outputs=[chat_output, chat_history])
     
     # Generate an image with Stable Diffusion
     generate_image_button.click(fn=generate_image, inputs=[image_input, image_model_input], outputs=image_output)