Update gradio_interface.py

gradio_interface.py

@@ -25,3 +25,24 @@ def gradio_chatbot(message, history, system_message, max_new_tokens, temperature
     )
     history.append((message, response))
     return history
+
+# Gradio interface
+def create_gradio_interface():
+    demo = gr.ChatInterface(
+        fn=gradio_chatbot,
+        additional_inputs=[
+            gr.Textbox(value=SYSTEM_MESSAGE, label="System message"),
+            gr.Slider(minimum=1, maximum=2048, value=DEFAULT_MAX_NEW_TOKENS, step=1, label="Max new tokens"),
+            gr.Slider(minimum=0.1, maximum=1.0, value=DEFAULT_TEMPERATURE, step=0.1, label="Temperature"),
+            gr.Slider(
+                minimum=0.1,
+                maximum=1.0,
+                value=DEFAULT_TOP_P,
+                step=0.05,
+                label="Top-p (nucleus sampling)",
+            ),
+        ],
+        title="Gemini Chat Interface",
+        description="Interact with the Gemini AI model."
+    )
+    return demo