mdSana commited on
Commit
75d5d7e
·
verified ·
1 Parent(s): cc6dad7

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +140 -0
app.py ADDED
@@ -0,0 +1,140 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import os
3
+ import urllib.request
4
+ from pydub import AudioSegment
5
+
6
+ # तीनों ओरिजिनल Piper मॉडल्स डाउनलोड और चेक करने का फंक्शन
7
+ def ensure_model_exists(voice_choice):
8
+ base_url = "https://huggingface.co/rhasspy/piper-voices/resolve/main/hi/hi_IN"
9
+
10
+ if voice_choice == "Female (Priyamvada)":
11
+ model_name = "hi_IN-priyamvada-medium"
12
+ url_path = f"{base_url}/priyamvada/medium"
13
+ elif voice_choice == "Male (Pratham)":
14
+ model_name = "hi_IN-pratham-medium"
15
+ url_path = f"{base_url}/pratham/medium"
16
+ else: # Male (Rohan)
17
+ model_name = "hi_IN-rohan-medium"
18
+ url_path = f"{base_url}/rohan/medium"
19
+
20
+ onnx_file = f"{model_name}.onnx"
21
+ json_file = f"{model_name}.onnx.json"
22
+
23
+ if not os.path.exists(onnx_file):
24
+ print(f"Downloading {onnx_file}...")
25
+ urllib.request.urlretrieve(f"{url_path}/{onnx_file}", onnx_file)
26
+ if not os.path.exists(json_file):
27
+ print(f"Downloading {json_file}...")
28
+ urllib.request.urlretrieve(f"{url_path}/{json_file}", json_file)
29
+
30
+ return onnx_file, model_name
31
+
32
+ # कैरेक्टर काउंट दिखाने का फंक्शन
33
+ def count_characters(text):
34
+ return f"📝 **अक्षरों की संख्या (Character Count):** {len(text)}"
35
+
36
+ def generate_news_audio(text, voice_choice, speed, apply_enhance):
37
+ # 1. टेक्स्ट क्लीनिंग: स्टार (**) और गैर-जरूरी शब्द ऑटोमैटिक हटाना
38
+ text = text.replace("मैं हूँ आपका न्यूज़ होस्ट", "")
39
+ text = text.replace("**", "")
40
+ text = text.replace("*", "")
41
+ text = text.strip()
42
+
43
+ if not text:
44
+ return None, "⚠️ कृपया टेक्स्ट दर्ज करें।"
45
+
46
+ try:
47
+ # 2. मॉडल सेटअप
48
+ onnx_file, model_name = ensure_model_exists(voice_choice)
49
+ raw_output = "raw_audio.wav"
50
+ final_output = "news_audio.wav"
51
+
52
+ # 3. स्पीड सेट करना (1.0 नॉर्मल है)
53
+ length_scale = 1.0 / float(speed)
54
+
55
+ # 4. वाक्य के बीच रुकने का समय (0.4 सेकंड)
56
+ sentence_silence = 0.4
57
+
58
+ # 5. Piper से Raw ऑडियो जनरेट करना
59
+ command = f'echo "{text}" | piper --model ./{onnx_file} --length_scale {length_scale} --sentence_silence {sentence_silence} --output_file {raw_output}'
60
+ os.system(command)
61
+
62
+ if os.path.exists(raw_output) and os.path.getsize(raw_output) > 0:
63
+
64
+ # --- स्टूडियो एनहांसमेंट लॉजिक (Studio Enhancement) ---
65
+ if apply_enhance:
66
+ # Raw ऑडियो को लोड करें
67
+ audio = AudioSegment.from_wav(raw_output)
68
+
69
+ # A. वॉल्यूम को लाउड और बैलेंस करें (Normalization)
70
+ audio = audio.normalize()
71
+
72
+ # B. रोबोटिक बेस और Hiss हटाने के लिए High-Pass Filter (80Hz)
73
+ audio = audio.high_pass_filter(80)
74
+
75
+ # एनहांस्ड ऑडियो को सेव करें
76
+ audio.export(final_output, format="wav")
77
+ else:
78
+ # अगर एनहांसमेंट बंद है, तो बिना बदलाव के फाइल का नाम बदल दें
79
+ os.rename(raw_output, final_output)
80
+
81
+ status_msg = f"✅ ऑडियो तैयार है! (Active Model: **{model_name}** | Enhanced: **{'Yes' if apply_enhance else 'No'}**)"
82
+ return final_output, status_msg
83
+ else:
84
+ return None, "❌ एरर: सर्वर ऑडियो जनरेट नहीं कर पाया।"
85
+
86
+ except Exception as e:
87
+ return None, f"❌ एरर: {str(e)}"
88
+
89
+ # --- UI Setup ---
90
+ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as app:
91
+ gr.Markdown("# 🎙️ Open-Source Hindi News TTS (Pro Studio)")
92
+
93
+ gr.Markdown("### ⚙️ **Available Models:** Piper TTS (`Priyamvada`, `Pratham`, `Rohan`)")
94
+ gr.Markdown("💡 **सुझाव:** मॉडल को सही से रुकने का निर्देश देने के लिए स्क्रिप्ट में **कॉमा (,)** और **फुल स्टॉप (। या .)** का इस्तेमाल ज़रूर करें। (स्क्रिप्ट पेस्ट करते ही स्टार ** ऑटोमैटिक हट जाएंगे)")
95
+
96
+ with gr.Row():
97
+ with gr.Column():
98
+ text_input = gr.Textbox(
99
+ label="न्यूज़ स्क्रिप्ट डालें",
100
+ lines=7,
101
+ placeholder="आज की मुख्य खबरें इस प्रकार हैं। भारत निर्वाचन आयोग ने..."
102
+ )
103
+
104
+ # यहाँ लाइव कैरेक्टर काउंट दिखेगा
105
+ char_count_display = gr.Markdown("📝 **अक्षरों की संख्या (Character Count):** 0")
106
+
107
+ voice_dropdown = gr.Dropdown(
108
+ choices=["Female (Priyamvada)", "Male (Pratham)", "Male (Rohan)"],
109
+ value="Female (Priyamvada)",
110
+ label="आवाज़ चुनें (Voice)"
111
+ )
112
+
113
+ speed_slider = gr.Slider(minimum=0.5, maximum=1.5, value=0.9, step=0.1, label="स्पीड (Speed) - न्यूज़ के लिए 0.9 या 1.0 सबसे सही है")
114
+
115
+ # --- नया एनहांसमेंट चेकबॉक्स ---
116
+ enhance_checkbox = gr.Checkbox(
117
+ label="🎙️ स्टूडियो एनहांसमेंट (Studio Enhancement) - आवाज़ को न्यूज़-रूम जैसी लाउड और क्लियर बनाने के लिए इसे चालू रखें",
118
+ value=True
119
+ )
120
+
121
+ generate_btn = gr.Button("Generate Audio", variant="primary")
122
+
123
+ with gr.Column():
124
+ status_output = gr.Markdown("स्टेटस यहाँ दिखेगा...")
125
+ audio_output = gr.Audio(label="जनरेटेड ऑडियो (यहाँ से डाउनलोड करें)", type="filepath")
126
+
127
+ # टेक्स्ट टाइप या पेस्ट करते ही कैरेक्टर काउंट अपडेट होगा
128
+ text_input.change(fn=count_characters, inputs=text_input, outputs=char_count_display)
129
+
130
+ # यहाँ स्पेसिंग की दिक्कत थी, जिसे अब ठीक कर दिया गया है
131
+ generate_btn.click(
132
+ fn=generate_news_audio,
133
+ inputs=[text_input, voice_dropdown, speed_slider, enhance_checkbox],
134
+ outputs=[audio_output, status_output],
135
+ api_name="predict"
136
+ )
137
+
138
+ if __name__ == "__main__":
139
+ app.launch()
140
+