Spaces:

Yakova
/

Embedding

Running

App Files Files Community

Embedding / gradio-app.py

Mbonea

fetcher = ProxyFetcher(loop=loop)

ab1d89e 3 months ago

raw history blame contribute delete

No virus

3 kB

	import gradio as gr
	import asyncio
	import copy
	import json
	import base64
	from io import BytesIO
	import IPython.display as ipd
	from watchgod import awatch
	from Complete import (
	Speak,
	Transcriber,
	MovieDescriptionProcessor,
	) # Replace 'your_script' with your actual module name

	tts_client = Speak()
	movie_processor = MovieDescriptionProcessor()


	# Gradio app function
	def gradio_app(movie_name, year, media_type, original_language, go_button):
	if go_button:
	print("here")
	# Construct the message for movie description
	message = movie_processor.prompt_fetcher()
	message = message.format(
	media_type=media_type,
	movie_name=movie_name,
	original_language=original_language,
	movie_year=year,
	)

	# Process movie description
	script = asyncio.run(movie_processor.process_movie_description(message))

	# Generate image descriptions
	message, result = movie_processor.prompt_fetcher("images").split("{format}")
	message = message.format(script=script) + result
	images_descriptions = asyncio.run(
	movie_processor.process_movie_description(message, "Assistant")
	)
	image_json = movie_processor.extract_json_from_markdown(images_descriptions)

	# Generate audio file
	audio_file_path = asyncio.run(tts_client.say(script))

	# Create an instance of the Transcriber class
	transcriber = Transcriber(audio_file_path)

	# Perform the transcription task
	asyncio.run(transcriber.transcribe())

	# Download the JSON transcription to a file
	output_json_path = "./output.json" # Replace with the desired output path
	asyncio.run(transcriber.download_transcription(output_json_path))

	# Return the outputs
	return {
	"image_json": image_json,
	"audio_file": {"name": "audio_file.wav", "file_path": audio_file_path},
	"tts_json": {"name": "tts_output.json", "file_path": output_json_path},
	}
	return None


	# Define Gradio inputs
	inputs = [
	gr.Textbox(placeholder="Avatar", label="Movie Name"),
	gr.Textbox(placeholder="2023", label="Year"),
	gr.Radio(choices=["movie", "tv series"], label="Media Type"),
	gr.Dropdown(choices=["English", "Spanish"], label="Original Language"),
	gr.Button("go_button"),
	]


	# Define Gradio outputs
	outputs = [
	gr.Textbox("image_json", label="Image JSON"),
	# gr.File("audio_file", label="Audio File"),
	# gr.File("tts_json", label="TTS JSON")
	]

	# Create the Gradio interface
	iface = gr.Interface(fn=gradio_app, inputs=inputs, outputs=outputs, live=True)


	# Launch the Gradio app
	async def restart_gradio():
	async for changes in awatch(".", watcher_cls=watchgod.DefaultDirWatcher):
	await iface.close()
	iface.launch()


	# Run Gradio interface and hot-reloading
	loop = asyncio.get_event_loop()
	loop.create_task(restart_gradio())
	iface.launch()