Update app.py

app.py

@@ -1,99 +1,44 @@
 import gradio as gr
-import sqlite3
 import torch
 import os
 import zipfile
 import requests
 from TTS.api import TTS
 
-# Create a database connection
-conn = sqlite3.connect("shared_models.db")
-cursor = conn.cursor()
-
-# Create a table to store shared models
-cursor.execute("""
- CREATE TABLE IF NOT EXISTS shared_models (
- id INTEGER PRIMARY KEY,
- model_url TEXT,
- model_name TEXT
- );
-""")
-
-# Define a function to store a shared model
-def store_shared_model(model_url, model_name):
- cursor.execute("INSERT INTO shared_models (model_url, model_name) VALUES (?,?)", (model_url, model_name))
- conn.commit()
-
-# Modify the get_shared_models function to display shared models
-def get_shared_models():
- cursor.execute("SELECT * FROM shared_models")
- models = cursor.fetchall()
- community_links.update(value="\n".join([f"{model[1]} ({model[0]})" for model in models]))
- return models
-
-# Define a function to clone a model
-def clone(text, url, language):
- # Store the shared model in the database
- store_shared_model(url, text)
- # Call the original clone function
- return original_clone(text, url, language)
-
-# Get the original clone function
-original_clone = clone
-
-# Modify the Textbox component to store shared models
-community_links = gr.components.Textbox(label="Community Links/Models")
-
-# Define a button to share a model
-share_button = gr.components.Button("Share Model")
-
-# Define a dropdown menu to retrieve shared models
-shared_models = get_shared_models()
-model_dropdown = gr.components.Dropdown(choices=[], label="Shared Models")
-
-# Define the clone function
-def clone_with_dropdown(text, url, language, model_id):
- # Retrieve the shared model from the database
- cursor.execute("SELECT * FROM shared_models WHERE id=?", (model_id,))
- model = cursor.fetchone()
- if model:
- # Call the clone function with the shared model URL
- return clone(text, model[1], language)
- else:
- # Call the original clone function
- return original_clone(text, url, language)
-
-# Update the interface
-iface = gr.Interface(fn=clone_with_dropdown,
- inputs=[
- gr.components.Text("", label="the text to be said"), 
- gr.components.Text("", label="URL of the zip file with the dataset on hf.co (10 seconds is fine)"),
- gr.Dropdown(choices=languages, label="Language"),
- community_links,
- share_button,
- model_dropdown
- ],
- outputs=gr.Audio(type='filepath'),
- title='Voice Clone',
- description="""
- by [Angetyde](https://youtube.com/@Angetyde?si=7nusP31nTumIkPTF) and [Tony Assi](https://www.tonyassi.com/ )
- use this colab with caution <3.
- """,
- theme=gr.themes.Base(primary_hue="teal", secondary_hue="teal", neutral_hue="slate"))
-
-# Define a function to handle the share button click event
-def handle_share_button(text, url, language):
- # Split the text into model name and URL
- model_name, model_url = text.split(",")
- # Store the shared model in the database
- store_shared_model(model_url, model_name)
- # Update the shared models list
- shared_models = get_shared_models()
- model_dropdown.update(choices=[(model[1], model[0]) for model in shared_models])
- # Clear the textbox
- community_links.update(value="")
-
-# Connect the share button to the handle_share_button function
-share_button.click(fn=handle_share_button, inputs=[community_links])
-
-iface.launch(share=True, on_launch=get_shared_models)
+#... (rest of the code remains the same)
+
+# Add a new section for model sharing
+model_sharing_section = gr.components.Section("Model Sharing")
+
+model_input = gr.components.Textbox(label="Share your model/link")
+model_output = gr.components.Markdown(label="Shared Models")
+
+model_sharing_iface = gr.Interface(
+ fn=lambda x: x, # dummy function to display the input
+ inputs=model_input,
+ outputs=model_output,
+ title="Share Your Model",
+ description="Share your model or link with the community!"
+)
+
+# Combine the two interfaces
+iface = gr.Interface(
+ fn=clone,
+ inputs=[
+ gr.components.Text("", label="the text to be said"), 
+ gr.components.Text("", label="URL of the zip file with the dataset on hf.co (10 seconds is fine)"), 
+ gr.Dropdown(choices=languages, label="Language")
+ ],
+ outputs=gr.Audio(type='filepath'),
+ title='Voice Clone',
+ description="""
+ by [Angetyde](https://youtube.com/@Angetyde?si=7nusP31nTumIkPTF) and [Tony Assi](https://www.tonyassi.com/ )
+ use this colab with caution <3.
+ """,
+ theme=gr.themes.Base(primary_hue="teal", secondary_hue="teal", neutral_hue="slate"),
+ tabs=["Voice Clone", "Model Sharing"]
+)
+
+iface.add_component(model_sharing_iface, "Model Sharing")
+
+iface.launch(share=True)