Spaces:
Running
Running
import os | |
import gradio as gr | |
import google.generativeai as genai | |
from gradio_client import Client, file | |
from dotenv import load_dotenv | |
# Load environment variables from .env file | |
load_dotenv() | |
# Retrieve API key from environment variable | |
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY") | |
# Retrieve system content from environment variable | |
SYSTEM_CONTENT = os.getenv("SYSTEM_CONTENT") | |
# Configure Google Gemini API | |
genai.configure(api_key=GEMINI_API_KEY) | |
# Create the model | |
generation_config = { | |
"temperature": 0.7, | |
"top_p": 0.95, | |
"top_k": 64, | |
"max_output_tokens": 512, # Adjust as needed | |
"response_mime_type": "text/plain", | |
} | |
# Define safety settings for the model | |
safety_settings = [ | |
{ | |
"category": "HARM_CATEGORY_HARASSMENT", | |
"threshold": "BLOCK_NONE" | |
}, | |
{ | |
"category": "HARM_CATEGORY_HATE_SPEECH", | |
"threshold": "BLOCK_NONE" | |
}, | |
{ | |
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", | |
"threshold": "BLOCK_NONE" | |
}, | |
{ | |
"category": "HARM_CATEGORY_DANGEROUS_CONTENT", | |
"threshold": "BLOCK_NONE" | |
} | |
] | |
# Create the generative model (outside the function) | |
model = genai.GenerativeModel( | |
model_name="gemini-1.5-pro", | |
generation_config=generation_config, | |
safety_settings=safety_settings, | |
system_instruction=SYSTEM_CONTENT, | |
) | |
# Initialize Gradio client for new TTS API (outside the function) | |
try: | |
tts_client = Client("tonyassi/voice-clone") | |
except ValueError as e: | |
print(f"Error initializing TTS client: {e}") | |
tts_client = None | |
def generate_response(user_input, chat_history): | |
"""Generates a response based on user input and chat history.""" | |
# Add user input to history | |
chat_history.append(user_input) | |
# Limit history length | |
if len(chat_history) > 10: | |
chat_history = chat_history[-10:] | |
# Start a new chat session | |
chat_session = model.start_chat() | |
# Send the entire chat history as the first message | |
response = chat_session.send_message("\n".join(chat_history)) | |
if tts_client: | |
# Use the new Gradio TTS API | |
tts_result = tts_client.predict( | |
text=response.text, | |
audio=file('audio.mp3'), # Use local audio file | |
api_name="/predict" | |
) | |
else: | |
tts_result = None | |
# Return response and audio, and update chat history | |
return response.text, tts_result, chat_history | |
# HTML code for the ad banner | |
ad_code = """ | |
<div style="text-align: center;"> | |
<iframe src="https://www.fiverr.com/gig_widgets?id=U2FsdGVkX1+9vQaosK0mq7xGXzIb/5hdFYIQsmHlpeMwFm5gingveDZKXBz2oJwV7ZwvToxseOx/KzKtr77ryB3qTb83/+GvCyI8OKpZcSgM0sDHzEPlZnqNeu8Db4y/IXlAceA1UcYT5Z3cW5vBLrgJ/WOpyMWSiyfwwMmGngjltQBexcivW7ukRd+/0yos7GCbfGjGdnLPHm2LB/CB9w2mA1i8cVv9LU+UUdj/O5KLRXtnychl9wAMkPTpmwMiyuAfUZy0nbK/Xa5O2UecmCvn7wicTr0TCTpLmIxaEh3YftXOVr9e36OGVYTmy5nlsgujkqJPI7wL0dXfon0ru4kvoycC7UD6m/whMoxrWZOf386qV2eEeXzk3vKO/emXUUfSavtofzSBtjLcyAOVhnHJg27PvVPgvFTQdy0o3F1M0DaHYTVW3Ln45MPYvMSH&affiliate_id=36184&strip_google_tagmanager=true" loading="lazy" data-with-title="false" class="fiverr_nga_frame" frameborder="0" height="350" width="100%" referrerpolicy="no-referrer-when-downgrade" data-mode="random_gigs" onload=" var frame = this; var script = document.createElement('script'); script.addEventListener('load', function() { window.FW_SDK.register(frame); }); script.setAttribute('src', 'https://www.fiverr.com/gig_widgets/sdk'); document.body.appendChild(script); " ></iframe> | |
</div> | |
""" | |
# HTML code for the ad banner | |
ad_code2 = """ | |
<div style="text-align: center;"> | |
<a href="https://beta.publishers.adsterra.com/referral/UNXJYTziBP" target="_blank" style="display: inline-block;"> | |
<img decoding="async" alt="banner" src="https://landings-cdn.adsterratech.com/referralBanners/gif/468x120_adsterra_reff.gif"> | |
</a> | |
<a href="https://go.fiverr.com/visit/?bta=36184&brand=fiverrcpa&landingPage=https%253A%252F%252Fwww.fiverr.com%252Fcategories%252Fprogramming-tech%252Fai-coding%252Fai-applications%253Fsource%253Dcategory_tree" target="_blank" style="display: inline-block;"> | |
<img fetchpriority="high" decoding="async" width="468" height="120" src="https://ziverr.xyz/wp-content/uploads/2024/06/PASSIVE-1.gif" class="attachment-large size-large wp-image-1266" alt=""> | |
</a> | |
</div> | |
""" | |
with gr.Blocks() as iface: | |
gr.HTML(ad_code) | |
gr.Interface( | |
fn=generate_response, | |
inputs=[ | |
gr.Textbox(lines=2, label="Chat with AI Donald Trump", placeholder="Enter your message here..."), | |
gr.State([]) # State input for chat history | |
], | |
outputs=[ | |
gr.Textbox(label="Response"), | |
gr.Audio(label="Voice Output", interactive=False, autoplay=True) if tts_client else gr.Textbox(label="Voice Output not available"), | |
gr.State([]) # State output to update chat history | |
], | |
title="AI Donald Trump", | |
description="Contact me if you want another character/voice<br>WhatsApp me: +92-332-4399819<br> Email me: aheedsajid@gmail.com<br><b>Donate something to increase GPU power</b><br>[Click here to Donate](https://nowpayments.io/donation/aheed)<br>Please duplicate the space if you get an error!" | |
) | |
gr.HTML(ad_code2) | |
iface.launch() | |