Update app.py

app.py

@@ -21,22 +21,47 @@ def generate_text(input_text, selected_model):
     output = model.generate(input_text, max_tokens=100)
     return output
 
-model_dropdown = gr.Dropdown(choices=model_choices(), 
-                             multiselect=False, 
-                             label="LLMs to choose from", 
-                             type="value", 
-                             value="orca-mini-3b-gguf2-q4_0.gguf")
+# model_dropdown = gr.Dropdown(choices=model_choices(), 
+#                              multiselect=False, 
+#                              label="LLMs to choose from", 
+#                              type="value", 
+#                              value="orca-mini-3b-gguf2-q4_0.gguf")
 
-explanation = gr.Textbox(label="Model Description", interactive=True, lines=10)
+# explanation = gr.Textbox(label="Model Description", interactive=True, lines=10)
 
-model_dropdown.change(fn=llm_intro, inputs=model_dropdown, outputs=explanation)
+# model_dropdown.change(fn=llm_intro, inputs=model_dropdown, outputs=explanation)
 
-input_text = gr.Textbox(lines=5, label="Input Text")
-output_text = gr.Textbox(lines=5, label="Generated Text")
+# input_text = gr.Textbox(lines=5, label="Input Text")
+# output_text = gr.Textbox(lines=5, label="Generated Text")
 
-gr.Interface(fn=generate_text, 
-             inputs=[input_text, model_dropdown], 
-             outputs=output_text, 
-             theme = gr.themes.Soft(), 
-             analytics_enabled=True, 
-             title="GPT4All Text Generation Experiment").launch()
+# gr.Interface(fn=generate_text, 
+#              inputs=[input_text, model_dropdown], 
+#              outputs=output_text, 
+#              theme = gr.themes.Soft(), 
+#              analytics_enabled=True, 
+#              title="GPT4All Text Generation Experiment").launch()
+
+with gr.Blocks() as demo:
+    gr.Markdown("## GPT4All Text Generation Experiment")
+    
+    # Dropdown to select model
+    model_selection = gr.Dropdown(label="Select an LLM", options=model_choices(), value=next(iter(model_description), None))
+    
+    # Textbox to show model description
+    explanation = gr.Textbox(label="Model Description", value="Model description", lines=10, readonly=True)
+    
+    # Link the dropdown with the textbox to update the description based on the selected model
+    model_selection.change(fn=llm_intro, inputs=model_selection, outputs=explanation)
+    
+    # Input text for generating text
+    input_text = gr.Textbox(lines=10, label="Input Text")
+    
+    # Textbox to display generated text
+    output_text = gr.Textbox(lines=10, label="Generated Text")
+    
+    # Button to generate text
+    generate_btn = gr.Button("Generate")
+    
+    generate_btn.click(fn=generate_text, inputs=[input_text, model_selection], outputs=output_text)
+
+demo.launch()