tushar310 commited on
Commit
70ca1f7
·
verified ·
1 Parent(s): 4af7cd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -40
app.py CHANGED
@@ -1,10 +1,7 @@
1
  import gradio as gr
2
  import os
3
- import uuid
4
  import time
5
- from io import BytesIO
6
  from openai import OpenAI
7
- from captcha.image import ImageCaptcha
8
 
9
  # --- Configuration ---
10
  client = OpenAI(
@@ -17,19 +14,6 @@ TITLE_HEADER = "AlphaAI Chatbot"
17
  MODEL_NAME = "deepseek/deepseek-r1-0528-qwen3-8b:free"
18
  ALPHAAI_CONTEXT = os.environ.get("ALPHAAI_CONTEXT", "")
19
 
20
- # --- Captcha State ---
21
- captcha_store = {}
22
-
23
- # --- Captcha Generation ---
24
- def generate_captcha():
25
- code = str(uuid.uuid4())[:6].upper()
26
- image = ImageCaptcha()
27
- buffer = BytesIO()
28
- image.write(code, buffer)
29
- buffer.seek(0)
30
- captcha_store[code] = True
31
- return buffer, code
32
-
33
  # --- Chatbot Backend ---
34
  def chat_with_llm(user_input):
35
  try:
@@ -48,26 +32,12 @@ def chat_with_llm(user_input):
48
 
49
  # --- Gradio UI ---
50
  with gr.Blocks() as demo:
51
- gr.Markdown("# 🤖 Alpha AI Assistant")
52
- gr.Markdown("Please solve the CAPTCHA to begin.")
53
-
54
- captcha_image = gr.Image(type="numpy", label="CAPTCHA Image")
55
- captcha_input = gr.Textbox(label="Enter CAPTCHA")
56
- verify_btn = gr.Button("Verify CAPTCHA")
57
- chatbot_area = gr.Column(visible=False)
58
 
59
- with chatbot_area:
60
- chatbot = gr.Chatbot(type="messages")
61
- msg = gr.Textbox()
62
- clear = gr.Button("Clear")
63
-
64
- def verify_captcha(user_code):
65
- user_code = user_code.strip().upper()
66
- if user_code in captcha_store:
67
- del captcha_store[user_code]
68
- return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
69
- else:
70
- return gr.update(), gr.update(value="Invalid CAPTCHA. Try again."), gr.update(), gr.update(visible=False)
71
 
72
  def user(user_message, history: list):
73
  return "", history + [{"role": "user", "content": user_message}]
@@ -84,14 +54,10 @@ with gr.Blocks() as demo:
84
  time.sleep(0.02)
85
  yield history
86
 
87
- image_data, correct_code = generate_captcha()
88
- captcha_image.value = image_data
89
-
90
- verify_btn.click(verify_captcha, inputs=[captcha_input], outputs=[captcha_image, captcha_input, captcha_input, chatbot_area])
91
  msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
92
  bot, chatbot, chatbot
93
  )
94
  clear.click(lambda: None, None, chatbot, queue=False)
95
 
96
  # Launch (required for Hugging Face Space)
97
- demo.launch()
 
1
  import gradio as gr
2
  import os
 
3
  import time
 
4
  from openai import OpenAI
 
5
 
6
  # --- Configuration ---
7
  client = OpenAI(
 
14
  MODEL_NAME = "deepseek/deepseek-r1-0528-qwen3-8b:free"
15
  ALPHAAI_CONTEXT = os.environ.get("ALPHAAI_CONTEXT", "")
16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  # --- Chatbot Backend ---
18
  def chat_with_llm(user_input):
19
  try:
 
32
 
33
  # --- Gradio UI ---
34
  with gr.Blocks() as demo:
35
+ gr.Markdown("# 🧠 Alpha AI Assistant")
36
+ gr.Markdown("Ask anything about Alpha AI.")
 
 
 
 
 
37
 
38
+ chatbot = gr.Chatbot(type="messages")
39
+ msg = gr.Textbox()
40
+ clear = gr.Button("Clear")
 
 
 
 
 
 
 
 
 
41
 
42
  def user(user_message, history: list):
43
  return "", history + [{"role": "user", "content": user_message}]
 
54
  time.sleep(0.02)
55
  yield history
56
 
 
 
 
 
57
  msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
58
  bot, chatbot, chatbot
59
  )
60
  clear.click(lambda: None, None, chatbot, queue=False)
61
 
62
  # Launch (required for Hugging Face Space)
63
+ demo.launch()